[ 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 无此参数，需要转写。 |
转写示例¶
alpha：source 的缩放倍数¶
# 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)
out：指定输出¶
# 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)