argsort

    对输入变量沿给定轴进行排序，输出排序好的数据的相应索引，其维度和输入相同。默认升序排列，如果需要降序排列设置 descending=True 。

    参数：

    返回：Tensor, 排序后索引信息（与 x 维度信息一致），数据类型为int64。

    代码示例：