2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.Sort.rst 1.14 KB
一键复制 编辑 原始数据 按行查看 历史
yeyunpeng2020 提交于 2024-02-04 16:10 . fix sort doc

mindspore.ops.Sort

.. py:class:: mindspore.ops.Sort(axis=-1, descending=False)

    根据指定的轴对输入Tensor的元素进行排序,默认为升序排序。

    .. warning::
        目前仅支持float16、uint8、int8、int16、int32、int64数据类型。如果使用float32类型可能导致数据精度损失。

    参数:
        - **axis** (int,可选) - 指定排序的轴。默认值: ``-1`` ,表示指定最后一维。当前Ascend后端只支持对最后一维进行排序。
        - **descending** (bool,可选) - 指定排序方式。如果 `descending` 为 ``True`` ,则根据value对元素进行降序排序。默认值: ``False`` 。

    输入:
        - **x** (Tensor) - 输入Tensor。

    输出:
        - **y1** (Tensor) - Tensor,其值为排序后的值,shape和数据类型与输入相同。
        - **y2** (Tensor) - 输入Tensor,其元素的索引。数据类型为int32。

    异常:
        - **TypeError** - `axis` 不是int。
        - **TypeError** - `descending` 不是bool。
        - **ValueError** - 当 `axis` 取值不在[-len(x.shape), len(x.shape))范围内。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助