• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# TopKFusion
2
3
4## 概述
5
6查找沿axis轴的前K个最大值和对应索引。
7
8该算子对应的[NodeType](_n_n_rt.md#nodetype)为NODE_TYPE_TOPK_FUSION。
9
10输入:
11
12- x,n维tensor。
13
14输出:
15
16- output0,axis维度的前K个最大值。
17
18- output1,axis维度的前K个最大值的索引。
19
20**Since:**
21
223.2
23
24**Version:**
25
261.0
27
28**相关模块:**
29
30[NNRt](_n_n_rt.md)
31
32
33## 汇总
34
35
36### Public 属性
37
38  | 名称 | 描述 |
39| -------- | -------- |
40| [sorted](#sorted) | boolean<br/>如果为True,按照大到小排序,如果为False,按照小到大排序。&nbsp; |
41| [axis](#axis) | long<br/>作用的轴&nbsp; |
42
43
44## 类成员变量说明
45
46
47### axis
48
49
50```
51long TopKFusion::axis
52```
53**描述:**
54
55作用的轴
56
57
58### sorted
59
60
61```
62boolean TopKFusion::sorted
63```
64**描述:**
65
66如果为True,按照大到小排序,如果为False,按照小到大排序。
67