\u200E

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

torch.Tensor.gather ¶

          torch.Tensor.gather(dim, index)

         

paddle.Tensor.take_along_axis ¶

          paddle.Tensor.take_along_axis(index, axis)

         

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

参数映射¶

PyTorch	PaddlePaddle	备注
dim	index	索引 Tensor，仅参数名不同。
index	axis	指定 index 获取输入的维度，仅参数名不同。