[ 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 无此参数,暂无转写方式。 |