Matlab 沿三维将矩阵与矩阵相乘

Question

我有一个大小为 m x n x 3 的矩阵 A，我有一个 3x3 矩阵 K。现在我想做的是这样的：

for row = 1:m
 for col = 1:n
  A(row,col,:) = K*[A(row,col,1);A(row,col,2);A(row,col,3)];
 end
end

我想要一个没有循环的有效解决方案，因为循环非常慢，因为 m x n 通常是图像的大小。

有人知道吗？

Answer 1

M = 1000;
N = 1000;
L = 3;
A = rand(M,N,L);
K = rand(L,L);
Q = reshape((K * reshape( A, [M*N, L] ).' ).', [M, N, L]);

错误检查：

Z = zeros(M,N,L);
for mm = 1 : M
  for nn = 1 : N
    Z(mm,nn,:) = K * squeeze( A(mm,nn,:) );
  end
end
max( abs( Z(:) - Q(:) ) )

ans =

      0

Matlab multiply matrix with matrix along third dimension