[torch 参数更多]torch.scatter¶

torch.scatter¶

torch.scatter(input,dim, index, src, reduce=None,out=None)

paddle.put_along_axis¶

paddle.put_along_axis(arr,indices, values, axis, reduce="assign", include_self=True, broadcast=True)

其中 Paddle 相比 PyTorch 支持更多其他参数，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ——- | ———— | ——- | | input | arr | 表示输入的 Tensor ，仅参数名不一致。 | | dim | axis | 表示在哪一个维度 scatter ，仅参数名不一致。 | | index | indices | 表示输入的索引张量，仅参数名不一致。 | | src | values | 表示需要插入的值，仅参数名不一致。 | | reduce | reduce | 归约操作类型 。 | | - | include_self | 表示插入 values 时是否包含 arr 中的元素，PyTorch 无此参数| | - | broadcast | 表示是否需要广播 indices 矩阵，PyTorch 无此参数，Paddle 应设置为 ‘False’ 结果才与 pytorch 一致。 | | out | - | 表示输出的 Tensor，Paddle 无此参数，需要转写。 |

转写示例¶

# Pytorch 写法
index = torch.tensor([[0],[1],[2]])
input = torch.zeros(3, 5)
out = torch.zeros(3, 5)
torch.scatter(input,1, index, 1.0,out=out)

# Paddle 写法
index = paddle.to_tensor(data=[[0], [1], [2]])
input = paddle.zeros(shape=[3, 5])
out = paddle.zeros(shape=[3, 5])
paddle.assign(paddle.put_along_axis(input, 1, index, 1.0), output=out)