mindspore.mint.argsort

mindspore.mint.argsort(input, dim=- 1, descending=False)

按指定顺序对输入Tensor沿给定维度进行排序，并返回排序后的索引。

警告

这是一个实验性API，后续可能修改或删除

参数：

• input (Tensor) - 待排序的输入Tensor。

• dim (int，可选) - 指定排序的轴。默认值：-1，表示指定最后一维。当前Ascend后端只支持对最后一维进行排序。

• descending (bool，可选) - 输出顺序。如果 descending 为 True ，按照元素值降序排序，否则升序排序。默认值： False 。

返回：

Tensor，排序后输入Tensor的索引。数据类型为int64。

支持平台：

Ascend

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> import mindspore.mint as mint
>>> x = Tensor(np.array([[8, 2, 1], [5, 9, 3], [4, 6, 7]]), mindspore.float16)
>>> sort = mint.argsort(x)
>>> print(sort)
[[2 1 0]
[2 0 1]
[0 1 2]]