[ 仅参数名不一致 ]torch.Tensor.argsort¶
torch.Tensor.argsort¶
torch.Tensor.argsort(dim=-1, descending=False, stable=False)
paddle.Tensor.argsort¶
paddle.Tensor.argsort(axis=-1, descending=False, stable=False, name=None)
其中 PyTorch 和 Paddle 功能一致,仅参数名不一致,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
dim | axis | 指定对输入 Tensor 进行运算的轴,仅参数名不一致。 |
descending | descending | 指定算法排序的方向。如果设置为 True,算法按照降序排序。如果设置为 False 或者不设置,按照升序排序。默认值为 False。 |
stable | stable | 是否使用稳定排序。 |