matlab 矩阵 三维矩阵相乘,将三维矩阵与二维矩阵相乘

我正在处理完全相同的问题,并着眼于最有效的方法。我看到的方法大致有三种,除了使用外部库(即,

mtimesx

):

循环通过三维矩阵的切片

重复和排列巫术

Cellfun乘法

我最近比较了这三种方法,看哪种方法最快。我的直觉是(2)会是赢家。代码如下:

% generate data

A = 20;

B = 30;

C = 40;

D = 50;

X = rand(A,B,C);

Y = rand(B,D);

% ------ Approach 1: Loop (via @Zaid)

tic

Z1 = zeros(A,D,C);

for m = 1:C

Z1(:,:,m) = X(:,:,m)*Y;

end

toc

% ------ Approach 2: Reshape+Permute (via @Amro)

tic

Z2 = reshape(reshape(permute(X, [2 1 3]), [A B*C]), [B A*C])' * Y;

Z2 = permute(reshape(Z2',[D A C]),[2 1 3]);

toc

% ------ Approach 3: cellfun (via @gnovice)

tic

Z3 = cellfun(@(x) x*Y,num2cell(X,[1 2]),'UniformOutput',false);

Z3 = cat(3,Z3{:});