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

样例：

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