文件最后提交记录最后更新时间
9 个月前
9 个月前
9 个月前
8 个月前
8 个月前
9 个月前
8 个月前
9 个月前
README

SortWithIndex

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT √

功能说明

  • 算子功能:将输入tensor按照元素值大小进行排序,index值跟随对应元素值进行排序。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待进行元素值排序的输入张量。 FLOAT16、FLOAT、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND
index 输入 待进行元素值排序的输入张量。 INT32 ND
y 输出 元素值排序后的输出结果。 FLOAT16、FLOAT、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND
sorted_index 输出 索引值随元素值排序后的输出结果。 INT32 ND

约束说明

-输入x和index的shape必须一致

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_sort_with_index 通过算子IR构图方式调用SortWithIndex算子。