!40035 add missing warning

Merge pull request !40035 from panzhihui/code_docs_update_sort
This commit is contained in:
i-robot 2022-08-09 09:33:26 +00:00 committed by Gitee
commit e3ac847504
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 1 deletions

View File

@ -5,6 +5,9 @@
根据指定的轴对输入Tensor的元素进行排序。默认为升序排序。
.. warning::
目前仅支持float16数据类型。如果使用float32类型可能导致数据精度损失。
参数:
- **axis** (int) - 指定排序的轴。默认值:-1。
- **descending** (bool) - 指定排序方式。如果 `descending` 为True则根据value对元素进行降序排序。默认值False。
@ -20,4 +23,4 @@
- **TypeError** - `axis` 不是int。
- **TypeError** - `descending` 不是bool。
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
- **ValueError** - 当 `axis` 取值不在[-len(x.shape), len(x.shape))范围内。
- **ValueError** - 当 `axis` 取值不在[-len(x.shape), len(x.shape))范围内。