mindspore.ops.diagonal_scatter

mindspore.ops.diagonal_scatter(input, src, offset=0, dim1=0, dim2=1)

将源tensor的值嵌入到输入tensor的对角线中（基于 dim1 和 dim2 指定的二维平面）。

说明

目前， input 和 src 中的元素不支持 inf 值。

参数：

• input (Tensor) - 维度大于1的输入tensor。

• src (Tensor) - 要嵌入的源tensor。

• offset (int, 可选) - 对角线偏移量。默认 0 。

  • 当 offset 是正整数时，对角线向上方偏移。

  • 当 offset 是负整数时，对角线向下方偏移。

• dim1 (int, 可选) - 确定指定平面对角线的第一维度。默认 0 。

• dim2 (int, 可选) - 确定指定平面对角线的第一维度。默认 1 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.ops.zeros((3, 3))
>>> output = mindspore.ops.diagonal_scatter(input, mindspore.ops.ones(3), 0)
>>> print(output)
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]
>>> output = mindspore.ops.diagonal_scatter(input, mindspore.ops.ones(2), 1)
>>> print(output)
[[0. 1. 0.]
 [0. 0. 1.]
 [0. 0. 0.]]