mindspore.ops.Diag
- class mindspore.ops.Diag[源代码]
用给定的对角线值构造对角线Tensor。
警告
这是一个实验性API,后续可能修改或删除。
更多参考详见
mindspore.ops.diag()
。- 输入:
input_x (Tensor) - 输入Tensor。
- 输出:
Tensor,数据类型与 input_x 一致。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> input_x = Tensor([1, 2, 3, 4]).astype('int32') >>> diag = ops.Diag() >>> output = diag(input_x) >>> print(output) [[1 0 0 0] [0 2 0 0] [0 0 3 0] [0 0 0 4]]