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

torch.Tensor.gather ¶

          torch.Tensor.gather(dim, index)

         

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

         

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

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