[ 参数不一致 ]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 。