[ 参数不一致 ]torch.sort¶

torch.sort¶

torch.sort(input,
           dim=-1,
           descending=False,
           stable=False,
           *,
           out=None)

paddle.sort¶

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

Pytorch 相比 Paddle 支持更多其他参数，同时两个 api 的返回参数类型不同，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 输入的 Tensor ，仅参数名不一致。 | | dim | axis | 指定对输入 Tensor 进行运算的轴。默认值为-1, 仅参数名不一致。 | | descending |descending | 指定算法排序的方向。如果设置为 True，算法按照降序排序。如果设置为 False 或者不设置，按照升序排序。默认值为 False，参数名相同。 | | stable | - | 使排序程序更稳定，保证等价元素的顺序得以保留。Paddle 无此参数，对于排序算法的稳定性来说不是很重要，直接删除即可。 | | out | - | 表示以(Tensor, LongTensor)输出的元组，含义是排序后的返回值和对应元素索引。Paddle 无此参数，若返回排序后的元素，需要转写；若需要返回元素和元素索引，需要结合 argsort 进行转写。 |

注：Pytorch 返回 (Tensor, LongTensor)，Paddle 返回 Tensor 。

转写示例¶

# 若要返回排序后的元素和元素索引，需要结合 argsort 进行转写
# Pytorch 写法
torch.sort(input, -1, True, out = (y, indices))

# Paddle 写法
paddle.assign(paddle.sort(input, -1, True), y)
paddle.assign(paddle.argsort(input, -1, True), indices)