MATLAB diag函数

MATLAB中的diag()函数有两种用法：

1. 将一个向量作为输入参数，返回一个以该向量为主对角线元素的方阵。

例如：将向量[1 2 3]输入diag()函数中，将返回一个3x3的方阵，其主对角线元素为1、2和3，其余元素为0。

A = diag([1 2 3])

输出结果为：

A =

     1     0     0
     0     2     0
     0     0     3

2. 将一个方阵作为输入参数，返回该方阵的主对角线元素构成的向量。

例如：将上面创建的方阵A输入diag()函数中，将返回一个向量[1 2 3]。

v = diag(A)

输出结果为：

v =

     1
     2
     3

需要注意的是，当输入参数为矩阵时，diag()函数只返回矩阵主对角线上的元素，而忽略其他元素。