paddle.sort 的接口能否对齐 pytorch

[sserdoubleh]

需求描述 Feature Description

任务目标（请描述你正在做的项目是什么，如模型、论文、项目是什么？）;
需求场景（请描述你的项目中为什么需要用此功能）;
功能描述（请简单描述或设计这个功能）

实现 top p sampling 的时候 利用到 paddle.sort & paddele.argsort 来获取 sorted_score & sorted_indices

improt paddle

probs = paddle.rand((3, 5))
sorted_probs = paddle.sort(probs, descending=True)
sorted_indices = paddle.argsort(probs, descending=True)

能否参考 pytorch：https://pytorch.org/docs/stable/generated/torch.sort.html?highlight=sort#torch.sort
直接同时提供两个结果

替代实现 Alternatives

import paddle
from paddle import _C_ops

probs = paddle.rand((3, 5))
sorted_probs, sorted_indices = _C_ops.argsort(probs, axis=-1, descending=True)