How can i vectorize a matrix multiplication of higher dimension arrays?
1 次查看(过去 30 天)
显示 更早的评论
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!
回答(2 个)
Steven Lord
2020-9-17
If you're using release R2020b or later, take a look at the pagemtimes function introduced in that release.
0 个评论
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Operating on Diagonal Matrices 的更多信息
产品
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!