[ 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 一致。 |