清空筛选条件
清空筛选条件

How to vectorize a for loop with array multiplication?

2 次查看（过去 30 天）
Erin Pratt
Erin Pratt 2021-6-23
评论： Erin Pratt 2021-6-23
Please help on replacing:
for i=1:Nm
Sloctemp(i,:,:)=Sloctemp(i,:,:)*Sort(m,i);
end
Where Nm = 19 Sloctemp is a 30x6x6 matrix, Sort is a 524288x19 matrix m=1
Sort is populated with only 0 or ones, so I am just trying to multiply Sloctemp(1:19,:,:) by either 1 or 0.
I tried:
i=1:Nm;
Sloctemp(i,:,:)=Sort(m,i).*Sloctemp(i,:,:);
but get the "Array dimensions must match for binary array op." error.
This section of the code gets run tens of millions of times, so I am just trying to get the fastes solution.
Thank you!

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

采纳的回答

Jan
Jan 2021-6-23
编辑：Jan 2021-6-23
Prefer to post the examples such, that they can be started by copy&paste.
Nm = 19;
Sloctemp = rand(30, 6, 6);
Sort = rand(524288, 19);
m = 1;
Result = Sloctemp;
Result(1:19, :, :) = Sloctemp(1:19, :, :) .* Sort(m, 1:19).';
% Compare with original version:
for i = 1:Nm
Sloctemp(i, :, :) = Sloctemp(i, :, :) * Sort(m, i);
end
max(abs(Result - Sloctemp), [], 'all')

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Shifting and Sorting Matrices 的更多信息

标签

产品


版本

R2016b

Community Treasure Hunt

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

Start Hunting!

Translated by