[ 仅 paddle 参数更多 ]torch.scatter_add

torch.scatter_add

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

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 | 表示插入 values 时的计算方式,PyTorch 无此参数,Paddle 应设置为 ‘add’ 。 | | - | include_self | 表示插入 values 时是否包含 arr 中的元素,PyTorch 无此参数,Paddle 应设置为 ‘True’ 。 | | - | broadcast | 表示是否需要广播 indices 矩阵,PyTorch 无此参数,Paddle 应设置为 ‘False’ 结果才与 pytorch 一致。 |