How to vectorize a for loop with array multiplication?
조회 수: 1 (최근 30일)
이전 댓글 표시
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!
댓글 수: 0
채택된 답변
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에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!