Hi, I want to speed up my script with double loop.

5 次查看（过去 30 天）
ArxIv
ArxIv 2024-4-5
评论： ArxIv 2024-4-6
I would like to eliminate double-loop that script below has.
I have difficulty in vectorizing/optimizing the code handling multi-dimensional arrays.
A is the sz×n matrix and B is the sz×m matrix. sz is the 1×p vector.
Thus, X shall be m×n×p matrix.
Can someone suggest me the
for i = 1:n
for j = 1:m
X(j,i,[1:length(sz)]) = A(:,i)' + B(:,j)';
end
end
I would be happy if some one suggest me solutions.

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

采纳的回答

Stephen23
Stephen23 2024-4-5
编辑：Stephen23 2024-4-5
X = permute(A,[3,2,1]) + permute(B,[2,3,1])
The data arrangement in memory is not optimal for that operation. Use BSXFUN for versions prior to R2016b.

更多回答（0 个）

类别

在 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