[ torch 参数更多 ]torch.Tensor.index_add¶
torch.Tensor.index_add¶
torch.Tensor.index_add(dim, index, source, *, alpha=1)
paddle.Tensor.index_add¶
paddle.Tensor.index_add(index, axis, value, name=None)
其中 PyTorch 与 Paddle 参数有差异,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
dim | axis | 表示进行运算的轴,仅参数名不一致。 |
index | index | 包含索引下标的 1-D Tensor。 |
source | value | 被加的 Tensor,仅参数名不一致。 |
alpha | - | source 的 缩放倍数, Paddle 无此参数,需要转写。Paddle 应将 alpha 和 source 的乘积作为 value。 |