[ 仅 paddle 参数更多 ]torch.scatter¶

torch.scatter¶

torch.scatter(input,
              dim,
              index,
              src)

paddle.put_along_axis¶

paddle.put_along_axis(arr,
                      indices,
                      values,
                      axis,
                      reduce='assign')

两者功能一致且参数用法一致，仅参数名不同，同时 Paddle 支持更多其他参数，具体如下：

参数差异¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | arr | 表示输入 Tensor ，仅参数名不一致。 | | dim | axis | 表示在哪一个维度 scatter ，仅参数名不一致。 | | index | indices | 表示输入的索引张量，仅参数名不一致。 | | src | values | 表示需要插入的值，仅参数名不一致。 | | - | reduce | 表示对输出 Tensor 的计算方式， PyTorch 无此参数， Paddle 保持默认即可。 |