harmony 鸿蒙TopKFusion
TopKFusion
概述
查找沿axis轴的前K个最大值和对应索引。
该算子对应的NodeType为NODE_TYPE_TOPK_FUSION。
输入:
- x,n维tensor。
输出:
output0,axis维度的前K个最大值。
output1,axis维度的前K个最大值的索引。
Since:
3.2
Version:
1.0
相关模块:
汇总
Public 属性
名称 | 描述 |
---|---|
sorted | boolean 如果为True,按照大到小排序,如果为False,按照小到大排序。 |
axis | long 作用的轴 |
类成员变量说明
axis
long TopKFusion::axis
描述:
作用的轴
sorted
boolean TopKFusion::sorted
描述:
如果为True,按照大到小排序,如果为False,按照小到大排序。
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类: 后端技术
- 本文标签:
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦