Matrix Multiplication Along Pages of Multidimensional Matrix

4 次查看（过去 30 天）
Mark Whirdy
Mark Whirdy 2014-8-27
评论： Mark Whirdy 2014-8-27
What is the most efficient way of matrix-multiplying along a page (2d plane) of a multidimensional matrix. Speed is important here as this operation will be in the objective-function of a calibration.
Specifically I have a 4d matrix of dimension [m,n,p,2], and I want to matrix-multiply it by a [2x1] vector
So conceptually I need
b*y0'
Taking for example:
b = rand(5,3,4,2); %
y0 = [0.1;0.2]; y0 = reshape(y0,[1,1,1,2]); %
I should end up with a 3d matrix of size [5,3,4]
I could, of course, dot-multiply by a repmatted y0 and then sum along the 4d but is this really the most efficient?
y0 = y0(ones(5,1,1),ones(1,3,1),ones(1,1,4),:);
sum(b.*y0,4);
All the best

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

采纳的回答

Matt J
Matt J 2014-8-27
If size(b,4)=2 all the time, it would probably be best just to generate two separate 3D arrays b1 and b2
result=b1*y0(1)+b2*y0(2);

更多回答（2 个）

Matt J
Matt J 2014-8-27
编辑：Matt J 2014-8-27
I could, of course, dot-multiply by a repmatted y0 and then sum along the 4d but is this really the most efficient?
Probably, if you're stuck with b in the shape you describe. However, it is better to use bsxfun, rather than repmat;
y0 = [0.1;0.2]; y0 = reshape(y0,[1,1,1,2]); %
result = sum( bsxfun(@times, b,y0) , 4)
  3 个评论
显示 1更早的评论
Matt J
Matt J 2014-8-27
It's not a meaningful test with m,n,p that small. If your dimensions are truly that size, you should be thinking about vectorizing across the different batches of b.

请先登录，再进行评论。

Matt J
Matt J 2014-8-27
编辑：Matt J 2014-8-27
If you can re-organize the code that generates b so that, without using permute() , it is instead 2 x m x n x p , that would be ideal. You can then do it all by straight matrix multiplication
result=squeeze( y0.'*reshape(b,2,[]) );

类别

在 Help Center 和 File Exchange 中查找有关 Matrix Indexing 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by