Extract a diagonal or construct a diagonal matrix.
If a is 1-D then a diagonal matrix is constructed with a on the
\(k\)-th diagonal. If a is 2-D then the \(k\)-th diagonal is
returned.
- 参数:
a (array) -- 1-D or 2-D input array.
k (int, optional) -- The diagonal to extract or construct.
Default: 0.
- 返回:
The extracted diagonal or the constructed diagonal matrix.
- 返回类型:
array
