\u200E

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

torch.Tensor.index_select ¶

          torch.Tensor.index_select(dim, index)

         

paddle.Tensor.index_select ¶

          paddle.Tensor.index_select(index, axis=0, name=None)

         

其中 PyTorch 和 Paddle 功能一致，仅参数名不一致，具体如下：

参数映射¶

PyTorch	PaddlePaddle	备注
dim	axis	索引轴，仅参数名不一致。
index	index	包含索引下标的 1-D Tensor。