[ torch 参数更多 ]torch.index_add¶

torch.index_add¶

torch.index_add(input, dim, index, source, *, alpha=1, out=None)

paddle.index_add¶

paddle.index_add(x, index, axis, value, name=None)

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

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 表示输入的 Tensor ，仅参数名不一致。 | | dim | axis | 表示进行运算的轴，仅参数名不一致。 | | index | index | 包含索引下标的 1-D Tensor。 | | source | value | 被加的 Tensor，仅参数名不一致。 | | alpha | - | source 的 缩放倍数， Paddle 无此参数，需要转写。 Paddle 应将 alpha 和 source 的乘积作为 value。| | out | - | 表示输出的 Tensor ， Paddle 无此参数，需要转写。 |

转写示例¶

# Pytorch 写法
torch.index_add(x, dim=1, index=index, source=source, alpha=alpha)

# Paddle 写法
paddle.index_add(x, index=index, axis=1, value=alpha*source)

# Pytorch 写法
torch.index_add(x, dim=1, index=index, source=source, alpha=alpha, out=y)

# Paddle 写法
paddle.assign(paddle.index_add(x, index=index, axis=1, value=alpha*source), y)