矢量化在嵌套疗法更高的维度为在Matlab环(Vectorizing ther higher dim

我有一个5D矩阵A ，我需要与载体乘以第3-5的尺寸。 例如，参见下面的示例代码：

A=rand(50,50,10,8,6);
B=rand(10,1);
C=rand(8,1);
D=rand(6,1);

for i=1:size(A,3)
    for j=1:size(A,4)
        for K=1:size(A,5)
            A(:,:,i,j,K)=A(:,:,i,j,K)*B(i)*C(j)*D(K);
        end
    end
end

我不知道是否有一个更好的\矢量\更快的方式做到这一点？

Answer 1:

首先，作为一个音符，这几天在Matlab，与JIT编译，矢量化代码不一定更快/更好。 对于大的问题，特别是内存使用可能会导致性能问题。

不过，这里要说的是似乎给了相同的结果作为你的代码向量化的解决方案：

A=rand(3,4,5,6,7);
B=rand(5,1);
C=rand(6,1);
D=rand(7,1);   

s=size(A);
[b,c,d]=ndgrid(B,C,D);
F=b.*c.*d;
G=zeros(1,1,prod(s(3:5)));
G(1,1,:)=F(:);
A=reshape(A,s(1),s(2),[]);
A=bsxfun(@times,A,G);
A=reshape(A,s);

编辑：另一种解决方案：

A=bsxfun(@times,A,permute(B,[2 3 1]));
A=bsxfun(@times,A,permute(C,[2 3 4 1]));
A=bsxfun(@times,A,permute(D,[2 3 4 5 1]));

文章来源: Vectorizing ther higher dimensions in nested for loop in Matlab