功能
后处理计算功能。实现输入tensor在最后一维上降序排列，并保留最大的num个元素，输出排序后的tensor及各元素对应的索引。
图1 SortOperation
约束
num是一个仅含有一个值的SVector，该值需大于0且小于等于输入x最后一维的大小。
定义
struct SortParam {
SVector<int32_t> num;
};
成员
成员名称 |
描述 |
|---|---|
num |
排序后保留的最大的元素的数量。 |
输入
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |
最后一维应至少有num个元素。 |
输出
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
output |
[-1,…,num] -1表示当前维度的大小没有约束。 |
float16 |
ND |
最后一维排序后，最大的num个元素。 |
indices |
[-1,…,num] -1表示当前维度的大小没有约束。 |
int32 |
ND |
最大的num个元素对应的原索引。 |