mindspore.mint.argsort ====================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/mint/mindspore.mint.func_argsort.rst :alt: 查看源文件 .. py:function:: mindspore.mint.argsort(input, dim=-1, descending=False, stable=False) 按指定顺序对输入Tensor沿给定维度进行排序,并返回排序后的索引。 .. warning:: 这是一个实验性API,后续可能修改或删除 参数: - **input** (Tensor) - 待排序的输入Tensor。 - **dim** (int,可选) - 指定排序的轴。默认值:``-1``,表示指定最后一维。当前Ascend后端只支持对最后一维进行排序。 - **descending** (bool,可选) - 输出顺序。如果 `descending` 为 ``True`` ,按照元素值降序排序,否则升序排序。默认值: ``False`` 。 - **stable** (bool,可选) - 是否使用稳定排序算法。 默认值: ``False`` 。 返回: Tensor,排序后输入Tensor的索引。数据类型为int64。 异常: - **ValueError** - 如果 `dim` 的设定值超出了范围。 - **TypeError** - 如果 `dim` 的数据类型不是int32。 - **TypeError** - 如果 `descending` 的数据类型不是bool。 - **TypeError** - 如果 `stable` 的数据类型不是bool。