提取矩阵的对角线元素

Question

我需要提取矩阵主对角线的元素：

我尝试了以下解决方案：

[U S V]= svd (T)
lambda= reshape(S',[],1);

但是我得到了包含所有元素的列向量，但是我只需要主对角线的元素

我也尝试过：

[U S V]= svd (T)
lambda = diag(S);

但这是行不通的

有人可以帮我提取对角线的元素吗？

Answer 1

您应该只可以直接在diag上使用S

lambda = diag(S);

从文档中

  

D = diag(v)返回一个正方形对角矩阵，在主对角线上有矢量v的元素。