mindspore.ops.diagonal_scatter
- mindspore.ops.diagonal_scatter(input, src, offset=0, dim1=0, dim2=1)[源代码]
dim1 和 dim2 指定 input 的两个维度,这两个维度上的元素将被视为矩阵的元素,并且将 src 嵌入到该矩阵的对角线上。
说明
目前,input 和 src 中的元素不支持
inf
值。- 参数:
input (Tensor) - 输入Tensor,其维度大于1。
src (Tensor) - 要嵌入的源Tensor。
offset (int, 可选) - 控制选择哪条对角线。可以是正值或负值。默认值:
0
。当 offset 是
0
时,选择的对角线是主对角线。当 offset 是正整数时,选择的对角线在主对角线上方。
当 offset 是负整数时,选择的对角线在主对角线下方。
dim1 (int, 可选) - 二维子数组的第一个轴,对角线应该从这里开始。默认值:
0
。dim2 (int, 可选) - 二维子数组的第二个轴,对角线应该从这里开始。默认值:
1
。
- 返回:
嵌入后的Tensor,具有与 input 相同的shape和dtype。
- 异常:
TypeError - input 或 src 不是Tensor。
TypeError - offset , dim1 或 dim2 不是整数。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore as ms >>> input = ms.ops.zeros((3,3)) >>> src = ms.ops.ones(2) >>> out = ms.ops.diagonal_scatter(input, src, 1, dim1=1, dim2=0) >>> print(out) [[0. 0. 0.] [1. 0. 0.] [0. 1. 0.]]