[仅参数名不一致]torch.Tensor.scatter_

torch.Tensor.scatter_

torch.Tensor.scatter_(dim, index, src)

paddle.Tensor.put_along_axis_

paddle.Tensor.put_along_axis_(index, value, axis, reduce="assign", include_self=True)

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

参数映射

PyTorch PaddlePaddle 备注
dim axis 表示在哪一个维度 scatter ,仅参数名不一致。
index indices 表示输入的索引张量,仅参数名不一致。
src values 表示需要插入的值,仅参数名不一致。
reduce reduce 归约操作类型,PyTorch 默认为 None, Paddle 保持默认即可。