How can i vectorize a matrix multiplication of higher dimension arrays?

5 次查看（过去 30 天）
Anthony Barone
Anthony Barone 2016-4-1
回答： Steven Lord 2020-9-17
I want to multiply a 6x6 matrix A and a 6x1 vector B at every grid point in a 3D earth model. Currently I am doing this:
for ix=1:length(x)
for iy=1:length(y)
for iz=1:length(z)
C(:,ix,iy,iz)=squeeze(A(:,:,ix,iy,iz))*squeeze(B(:,ix,iy,iz));
end
end
end
Is there a way to vectorize this? Currently this segment of nested loops is by far the slowest part of my code (almost 90% of the time taken every iteration is spent in this nested loop). Thanks!
  1 个评论
Anthony Barone
Anthony Barone 2016-4-1
I have found an answer, but I still wanted to post it incase anyone else has a similiar situation. I did this:
for n=1:6
C(n,:,:,:)=sum(squeeze(A(n,:,:,:,:)).*B);
end
For me, this method proved to be ~30x faster than my initial method.

请先登录，再进行评论。

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

回答（2 个）

Steven Lord
Steven Lord 2020-9-17
If you're using release R2020b or later, take a look at the pagemtimes function introduced in that release.
Matt J
Matt J 2016-4-1
Using mtimesx ( Download )
C=mtimesx( reshape(A,6,6,[]) , reshape(B,6,1,[]) );
C=reshape(C,size(B));

类别

在 Help Center 和 File Exchange 中查找有关 Just for fun 的更多信息

标签

产品

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by