Matlab：N維數組對角化形成diag矩陣

Question

我需要從 3D 數組形成對角矩陣：

給定A =rand(2,2,5) 。 B是一個矩陣，其對角線元素是A切片

A具有以下切片：a1、a2、a3、a4 和 a5，切片 2 x 2。

B應如下所示：

[a1 0 0 0 0;
 0 a2 0 0 0;
 0 0 a3 0 0;
 0 0 0 a4 0;
 0 0 0 0 a5]

我已將其實現為：

lambda=[];

for m=1:N
    l=A(:,:,m);
    lambda = [lambda; l];

end

它將矩陣形成為 10 x 2。我的錯誤在哪里？

Answer 1

有一個內置函數： blkdiag ：

% random matrice
A = rand(2,2,5)
% split A into 2D matrix (or slice) we preserve the 1st and 2nd dimension
C = num2cell(A,[1 2])
% block diagonalize
B = blkdiag(C{:})