• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# TopKFusion
2
3## 概述
4
5查找沿axis轴的前K个最大值和对应索引。
6
7该算子对应的[NodeType](_n_n_rt_v20.md#nodetype)为NODE_TYPE_TOPK_FUSION。
8
9输入:
10
11- x,n维tensor。
12
13输出:
14
15- output0,axis维度的前K个最大值。
16
17- output1,axis维度的前K个最大值的索引。
18
19**起始版本:** 3.2
20
21**相关模块:**[NNRt](_n_n_rt_v20.md)
22
23## 汇总
24
25### Public 属性
26
27| 名称 | 描述 |
28| -------- | -------- |
29| boolean [sorted](#sorted) | 如果为True,按照大到小排序,如果为False,按照小到大排序。  |
30| long [axis](#axis) | 作用的轴。  |
31
32## 类成员变量说明
33
34### axis
35
36```
37long TopKFusion::axis
38```
39**描述**
40
41作用的轴。
42
43### sorted
44
45```
46boolean TopKFusion::sorted
47```
48**描述**
49
50如果为True,按照大到小排序,如果为False,按照小到大排序。
51