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