[仅参数名不一致]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 保持默认即可。 |