mindspore.Tensor.diag

Constructs a diagonal tensor with a given diagonal values.

Assume self tensor has dimensions $[D_{1}, . . . D_{k}]$ , the output is a tensor of rank 2k with dimensions $[D_{1}, . . ., D_{k}, D_{1}, . . ., D_{k}]$ where: $o u t p u t [i_{1}, . . ., i_{k}, i_{1}, . . ., i_{k}] = s e l f [i_{1}, . . ., i_{k}]$ and 0 everywhere else.

Returns: Tensor, has the same dtype as self tensor.
Raises: ValueError – If rank of self tensor is less than 1.

Supported Platforms:: Ascend GPU

Examples

>>> from mindspore import Tensor
>>> x = Tensor([1, 2, 3, 4]).astype('int32')
>>> output = x.diag()
>>> print(output)
[[1 0 0 0]
 [0 2 0 0]
 [0 0 3 0]
 [0 0 0 4]]