mindspore.ops.diagflat

mindspore.ops.diagflat(input, offset=0)

Create a 2-D Tensor which diagonal is the flattened input .

Parameters

• input (Tensor) – Input Tensor, which is flattened and set as the diagonal of the output.

• offset (int, optional) –

  offset controls which diagonal to choose. Default: 0 .

  • When offset is zero, the diagonal chosen is the main diagonal.

  • When offset is a positive integer, the diagonal chosen is up the main diagonal.

  • When offset is a negative integer, the diagonal chosen is down the main diagonal.

Returns

The 2-D Tensor, whose diagonal is the flattened input.

Raises

• TypeError – If input is not a tensor.

• TypeError – If offset is not an integer.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> x = Tensor([1, 2], mindspore.float32)
>>> output = ops.diagflat(x, 1)
>>> print(output)
[[0. 1. 0.]
 [0. 0. 2.]
 [0. 0. 0.]]