mindspore.ops.Diag

View Source On Gitee
class mindspore.ops.Diag[source]

Constructs a diagonal tensor with a given diagonal values.

Warning

This is an experimental API that is subject to change or deletion.

Refer to mindspore.ops.diag() for more details.

Inputs:

  • input_x (Tensor) - The input tensor.

Outputs:

Tensor, has the same dtype as the input_x.

Supported Platforms:

Ascend GPU CPU

Examples

>>> from mindspore import Tensor, ops
>>> 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]]