Vectorized nD diag loop

Question

Javier Fernandez 2022-7-5

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1753450-vectorized-nd-diag-loop

评论： Javier Fernandez 2022-7-5

采纳的回答： Chunru

在 MATLAB Online 中打开

Hello everyone,

I have the following loop which I would want to have vectorized.

I know the diag() command cannot be used in nD arrays (n>2), so is there an alternative way to vectorize the loop?

n = 10; m = 15;
A = rand(4,n,m);
for i = 1:n
    for j = 1:m
B(:,:,i,j) = diag(A(:,i,j));
    end
end

Thanks in advance!

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Chunru 2022-7-5

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1753450-vectorized-nd-diag-loop#answer_1000080

在 MATLAB Online 中打开

n = 400; m = 500;
A = rand(4,n,m);
B = zeros(4,4,n,m);   % initialize to speed up
tic
for i = 1:n
    for j = 1:m
        B(:,:,i,j) = diag(A(:,i,j));
    end
end
toc
Elapsed time is 0.351235 seconds.
% Vectorized
tic
C = zeros(4*4,n*m);
C(1:5:16, :) = A(1:4, :);
C =reshape(C, [4, 4, n, m]);
toc
Elapsed time is 0.015569 seconds.
isequal(B, C)
ans = logical
   1