mindspore.ops.argsort

mindspore.ops.argsort(input, axis=- 1, descending=False)

返回按指定轴对tensor进行排序后的索引。

说明

当前Ascend后端只支持对最后一维进行排序。

参数：

• input (Tensor) - 输入tensor。

• axis (int) - 指定排序的轴。默认 -1 。

• descending (bool) - 指定排序（升序或降序）。默认 False 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> # case 1: 1-dimensional sort
>>> input = mindspore.tensor([1, 3, 5, 4, 2, 1])
>>> mindspore.ops.argsort(input)
Tensor(shape=[6], dtype=Int32, value= [0, 5, 4, 1, 3, 2])
>>>
>>> # case 2: multi-dimensional sort
>>> input = mindspore.tensor([[2, 1, 3],
...                           [6, 4, 3]])
>>> mindspore.ops.argsort(input, axis=1)
Tensor(shape=[2, 3], dtype=Int32, value=
[[1, 0, 2],
 [2, 1, 0]])
>>> mindspore.ops.argsort(input, axis=1, descending=True)
Tensor(shape=[2, 3], dtype=Int32, value=
[[2, 0, 1],
 [0, 1, 2]])