mindspore.numpy.diag
- mindspore.numpy.diag(v, k=0)[source]
Extracts a diagonal or construct a diagonal array.
- Parameters
- Returns
Tensor, the extracted diagonal or constructed diagonal array.
- Raises
ValueError – if input is not 1-D or 2-D.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore.numpy as np >>> x = np.arange(9).reshape((3,3)) >>> print(x) [[0 1 2] [3 4 5] [6 7 8]] >>> output = np.diag(x) >>> print(output) [0 4 8] >>> output = np.diag(x, k=1) >>> print(output) [1 5] >>> output = np.diag(x, k=-1) >>> print(output) [3 7]