Sort

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：将输入tensor中的元素根据指定维度进行升序/降序排序，并且返回对应的排序结果值和索引值。

• 计算公式：

$$y1=\mathrm{sort}(x,\text{axis},\text{descending},\text{stable})$$

$$y2=\mathrm{argsort}(x,\text{axis},\text{descending},\text{stable})$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行排序的输入tensor。	FLOAT16、FLOAT32、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BFLOAT16	ND
y1	输出	表示tensor在指定维度上排序的结果值，与x具有相同的类型和格式。	FLOAT16、FLOAT32、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BFLOAT16	ND
y2	输出	表示排序后每个元素在原tensor中的索引。	INT32、INT64	ND
axis	属性	指定排序的维度，默认为-1（最后一维）。	INT	-
descending	属性	是否降序排序，true为降序，false为升序，默认为false。	BOOL	-
stable	属性	是否稳定排序，true为稳定排序，false为非稳定排序，默认为false。	BOOL	-
y2_dtype	属性	y2的数据类型，默认为DT_INT32。	Type	-

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_sort	通过aclnnSort接口方式调用Sort算子。
aclnn调用	test_aclnn_argsort	通过aclnnArgsort接口方式调用Sort算子。