[ paddle 参数更多 ]torch.Tensor.gather

torch.Tensor.gather

torch.Tensor.gather(dim, index)

paddle.Tensor.take_along_axis

paddle.Tensor.take_along_axis(indices, axis, broadcast=True)

其中 Paddle 相比 PyTorch 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
dim axis 指定 index 获取输入的维度,仅参数名不一致。
index indices 索引 Tensor,仅参数名不一致。
- broadcast 表示是否需要广播 indices 矩阵,PyTorch 无此参数,Paddle 应设置为 False 结果才与 pytorch 一致。