matlab三维向量,第三维向量的Matlab产品

Matlab中的循环不再是一件令人恐惧的事情,也不再是本身所避免的事情.

当然,在使用它们时应该非常小心,但是,JIT可以处理多种循环,甚至超越内置函数也能提高性能.

考虑以下测试用例：

clc

m = 512; n = 384;

p = 5; q = 3;

A = rand(m,n,p); % some sample data

B = rand(m,n,q); % some sample data

%% non-loop approach

tic

A2 = reshape(A,[],p);

B2 = reshape(B,[],q);

C2 = arrayfun(@(ii) A2(ii,:)'*B2(ii,:),1:m*n,'uni',false);

C0 = permute(reshape(cell2mat(C2),p,q,m,n),[3 4 1 2]);

toc

%% looped approach, simplest

tic

C = zeros(m,n,p,q);

for mm = 1:m

for nn = 1:n

C(mm,nn,:,:) = ...

squeeze(A(mm,nn,:))*squeeze(B(mm,nn,:)).';

end

end

toc

% check for equality

all(C0(:) == C(:))

%% looped approach, slightly optimized

tic

C = zeros(m,n,p,q);

pp = zeros(p,1);

qq = zeros(1,q);

for mm = 1:m

for nn = 1:n

pp(:) = A(mm,nn,:);

qq(:) = B(mm,nn,:);

C(mm,nn,:,:) = pp*qq;

end

end

toc

% check for equality

all(C0(:) == C(:))

%% looped approach, optimized

tic

C = zeros(p,q,m*n);

A2 = reshape(A,[],p);

B2 = reshape(B,[],q);

for mn = 1:m*n

C(:,:,mn) = A2(mn,:).'*B2(mn,:);

end

C = permute(reshape(C, p,q,m,n), [3,4,1,2]);

toc

% check for equality

all(C0(:) == C(:))

结果：

Elapsed time is 3.955728 seconds.

Elapsed time is 21.013715 seconds.

ans =

1

Elapsed time is 1.334897 seconds.

ans =

1

Elapsed time is 0.573624 seconds.

ans =

1

无论性能如何,我还发现最后一种情况比非循环情况更直观和可读.