[ 仅参数名不一致 ]torch.Tensor.index_fill_

torch.Tensor.index_fill_

torch.Tensor.index_fill_(dim, index, value)

paddle.Tensor.index_fill_

paddle.Tensor.index_fill_(index, axis, value, name=None)

两者功能一致且参数用法一致,仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
dim axis 表示进行运算的轴,仅参数名不一致。
index index 包含索引下标的 1-D Tensor。
value value 填充的值。

转写示例

alpha:source 的缩放倍数

# Pytorch 写法
x.index_fill_(dim=1, index=index, value=1)

# Paddle 写法
x.index_fill_(index=index, axis=1, value=1)