mindspore.ops.argsort
- mindspore.ops.argsort(input, axis=- 1, descending=False)[源代码]
按指定顺序对输入Tensor沿给定维度进行排序,并返回排序后的索引。
- 参数:
input (Tensor) - 待排序的输入Tensor。
axis (int) - 指定排序的轴。默认值:
-1
,表示指定最后一维。当前Ascend后端只支持对最后一维进行排序。descending (bool) - 输出顺序。如果 descending 为
True
,按照元素值降序排序,否则升序排序。默认值:False
。
- 返回:
Tensor,排序后输入Tensor的索引。数据类型为int32。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> x = Tensor(np.array([[8, 2, 1], [5, 9, 3], [4, 6, 7]]), mindspore.float16) >>> sort = ops.argsort(x) >>> print(sort) [[2 1 0] [2 0 1] [0 1 2]]