Array multiplication along certain dimension?

Question

Visa Suomi 2017-11-21

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/368379-array-multiplication-along-certain-dimension

评论： Visa Suomi 2017-11-28

I have a 4-D array with dimensions A = A1 x A2 x A3 x N and a vector with dimensions V = 1 x N. I would like to multiply each element in the array A with vector V along fourth dimension and take a sum along the same dimension to form a matrix B = A1 x A2 x A3. Currently I am using a loop to achieve this, but is there any faster method to achieve the same result? Thanks!

A1 = 3;
A2 = 3;
A3 = 3;
N = 10;
A = ones(A1, A2, A3, N);
V = ones(1, N);
B = NaN(A1, A2, A3);
for i = 1:A1
for j = 1:A2
for k = 1:A3
  B(i,j,k) = squeeze(A(i,j,k,:))' * V';
end
end
end

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

David Goodmanson 2017-11-24

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/368379-array-multiplication-along-certain-dimension#answer_292930

编辑：David Goodmanson 2017-11-24

在 MATLAB Online 中打开

Hi Visa,

This seems to work

A1 = 2;
A2 = 3;
A3 = 4;
N =  7;
A = rand(A1, A2, A3, N);
V = rand(1, N);
% ---- method 1
B1 = NaN(A1, A2, A3);
for i = 1:A1
  for j = 1:A2
    for k = 1:A3
      B1(i,j,k) = squeeze(A(i,j,k,:))' * V';
    end
  end
end
% ---- method 2
D = permute(A,[4 1 2 3]);
E = reshape(D,N,A1*A2*A3);
F = V*E;
B2 = reshape(F,A1,A2,A3);
max(max(max(max(abs(B1-B2)))))     ans = 8.8818e-16