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

torch.Tensor.index_fill ¶

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

         

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

         

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

           # PyTorch 写法
x.index_fill(dim=1, index=index, value=1)

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