[ torch 参数更多 ]torch.argsort¶

torch.argsort¶

torch.argsort(input, dim=- 1, descending=False, stable=False)

paddle.argsort¶

paddle.argsort(x, axis=- 1, descending=False, name=None)

其中 Pytorch 相比 Paddle 支持更多其他参数，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 输入的多维 Tensor ，仅参数名不同。 | | dim | axis | 指定进行运算的轴，仅参数名不同。 | | descending | descending | 是否使用降序排列。 | | stable | - | 是否使用稳定排序。Paddle 无此参数，暂无转写方式。 |