mindspore.ops.argsort

mindspore.ops.argsort(input, axis=- 1, descending=False)[源代码]

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

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

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

  • descending (bool) - 输出顺序。如果 descendingTrue ,按照元素值降序排序,否则升序排序。默认值: 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]]