mindspore.ops.diagflat

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

创建一个二维Tensor，用展开后的 input 作为它的对角线。

参数：

• input (Tensor) - 输入Tensor，展开后作为输出的对角线。

• offset (int, 可选) - offset 控制选择哪条对角线。默认值： 0 。

  • 当 offset 是0时，选择的对角线是主对角线。

  • 当 offset 是正整数时，选择的对角线在主对角线上方。

  • 当 offset 是负整数时，选择的对角线在主对角线下方。

返回：

二维Tensor，对角线是展开的 input 。

异常：

• TypeError - input 不是Tensor。

• TypeError - offset 不是整数。

支持平台：

Ascend GPU CPU

样例：

>>> 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.]]