Further vectorizing a loop over bsxfun

9 次查看（过去 30 天）
Niko
Niko 2011-11-5
I have a vectorization problem that I cannot quite wrap my head around. I have a calculation that requires several parameter sweeps.
Basically, I need to element-wise multiply a matrix A of size m*n and B matrix of size m*p to obtain a matrix Y of size m*n*p. My solution so far is
for i=1:length(B)
Y(:,:,i)=bsxfun(@times,A,B(:,i));
end
This strikes me as unnecessary, surely there must be a way to generalize this to higher dimensions without using loops? Another solution I have cooked up uses repmat, but seems even less elegant.
I hope someone can point into the right direction :)
  3 个评论
显示 1更早的评论
Daniel Shub
Daniel Shub 2011-11-5
I am assuming your code is really i=1:size(B, 2), otherwise I get an error.
Niko
Niko 2011-11-5
Indeed, I made a mistake when reducing to a minimal working example. Thank you.

请先登录，再进行评论。

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

采纳的回答

Andrei Bobrov
Andrei Bobrov 2011-11-5
bsxfun(@times,A,reshape(B,size(B,1),1,[]))
or
bsxfun(@times,A,permute(B,[1 3 2]))

更多回答（0 个）

类别

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

标签

产品

Community Treasure Hunt

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

Start Hunting!

Translated by