| Files | Last commit | Last update |
|---|---|---|
| 1 month ago | ||
| 10 months ago | ||
| 8 months ago | ||
| 1 month ago | ||
| 2 months ago | ||
| 3 months ago | ||
| 1 month ago | ||
| 1 month ago | ||
| 10 days ago |
SortWithIndex
English | 简体中文
Product Support Status
| Product | Supported |
|---|---|
| Ascend 950PR&950DT products | √ |
| Atlas A3 products | × |
| Atlas A2 products | × |
| Atlas 200I/500 A2 inference products | × |
| Atlas inference products | × |
| Atlas training products | √ |
Function Description
- Operator function: sorts the input tensor by element value, and sorts the index values along with the corresponding element values.
Parameter Description
| Parameter Name | Input/Output/Attribute | Description | Data Type | Data Format |
|---|---|---|---|---|
| x | Input | Input tensor for element value sorting. | FLOAT16, FLOAT, BFLOAT16, INT8, INT16, INT32, INT64, UINT8, UINT16, UINT32, UINT64 | ND |
| index | Input | Input tensor for element value sorting. | INT32, INT64 | ND |
| axis | Input | Dimension used as the sorting criterion. | INT64 | - |
| descending | Input | Controls the sorting order. True indicates descending order, and False indicates ascending order. | BOOL | - |
| stable | Input | Whether to perform stable sorting. True indicates stable sorting, and False indicates unstable sorting. | BOOL | - |
| y | Output | Output result after elements are sorted by value. | FLOAT16, FLOAT, BFLOAT16, INT8, INT16, INT32, INT64, UINT8, UINT16, UINT32, UINT64 | ND |
| sorted_index | Output | Output result after the index values are sorted along with the element values. | INT32, INT64 | ND |
Constraint Description
- The shapes of input x and index must be the same.
Invocation Description
| Invocation Method | Invocation Sample | Description |
|---|---|---|
| Graph-mode invocation | test_geir_sort_with_index | Invokes the SortWithIndex operator by building a graph through the operator IR. |