matlab矩阵矢量乘法,matlab – 矢量化矩阵乘法

将B重塑为2D,使用A执行矩阵乘法以减小B的第一维和A的第二维,并获得2D乘积输出.最后将产品重新塑造成所需的3D输出,就像这样 –

[m,n,p] = size(B); %// Store size parameters

C = reshape(A*reshape(B,m,[]),m,n,p)

标杆

num_iter = 100; %// Number of iterations to run benchmarks

for k = 1:50000

tic(); elapsed = toc(); %// Warm up tic/toc.

end

%// Size parameters and setup input arrays

m = 14;

n = 256;

p = 200;

A = rand(m,m);

B = rand(m,n,p);

disp('---------------- With loopy approach')

tic

for ii = 1:num_iter

C = zeros(m,n,p);

for i = 1:p

C(:, :, i) = A*B(:,:,i);

end

end

toc

disp('---------------- With vectorized approach')

tic

for ii = 1:num_iter

[m,n,p] = size(B);

Cout = reshape(A*reshape(B,m,[]),m,n,p);

end

toc

error_check = isequal(C,Cout) %// 1 means good

输出 –

---------------- With loopy approach

Elapsed time is 1.679919 seconds.

---------------- With vectorized approach

Elapsed time is 1.496923 seconds.

error_check =

1