harmony 鸿蒙TopKFusion
TopKFusion
概述
查找沿axis轴的前K个最大值和对应索引。
该算子对应的NodeType为NODE_TYPE_TOPK_FUSION。
输入：
- x，n维tensor。
输出：
output0，axis维度的前K个最大值。
output1，axis维度的前K个最大值的索引。
起始版本： 3.2
相关模块：NNRt
汇总
Public 属性
| 名称 | 描述 |
|---|---|
| boolean sorted | 如果为True，按照大到小排序，如果为False，按照小到大排序。 |
| long axis | 作用的轴。 |
类成员变量说明
axis
long TopKFusion::axis
描述
作用的轴。
sorted
boolean TopKFusion::sorted
描述
如果为True，按照大到小排序，如果为False，按照小到大排序。
你可能感兴趣的鸿蒙文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
7、 openharmony
-
9、 golang