matlab diag

在 MATLAB 中，diag 函数可以用来获取矩阵的对角线元素或者将向量转换为对角矩阵。

获取矩阵的对角线元素：

使用方式：diag(A)

假设 A 是一个 n 行 m 列的矩阵，那么 diag(A) 返回一个 min(n,m) 行 1 列的向量，包含了 A 的主对角线上的元素。

例如，对于一个 3 行 4 列的矩阵 A：

A = [1 2 3 4;
     5 6 7 8;
     9 10 11 12];

diag(A) 的结果是一个 3 行 1 列的向量：

ans = [1;
       6;
       11];

将向量转换为对角矩阵：

使用方式：diag(v)

假设 v 是一个 n 行 1 列的向量，那么 diag(v) 返回一个 n 行 n 列的对角矩阵，对角线上的元素就是 v 的元素。

例如，对于一个 3 行 1 列的向量 v：

v = [1;
     2;
     3];

diag(v) 的结果是一个 3 行 3 列的对角矩阵：

ans = [1 0 0;
       0 2 0;
       0 0 3];