FilesLast commitLast 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
README

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.