1# ArgMaxFusion
2
3
4## 概述
5
6返回跨轴的tensor前K个索引或者是数值。
7
8该算子对应的[NodeType](_n_n_rt_v10.md#nodetype)为NODE_TYPE_ARGMAX_FUSION。
9
10输入：
11
12- x，n维tensor，输入张量（N,\*）,其中\*意味着数量任意的附加维度。
13
14输出：
15
16- output，轴上输入张量最大值的前K个索引或者是数值。
17
18**起始版本：** 3.2
19
20**相关模块：**[NNRt](_n_n_rt_v10.md)
21
22
23## 汇总
24
25
26### Public 属性
27
28| 名称 | 描述 |
29| -------- | -------- |
30| long [axis](#axis) | 指定求最大值索引的维度。  |
31| long [topK](#topk) | 轴上前K个最大值。  |
32| boolean [keepDims](#keepdims) | 是否保持输出维度和输入的维度是否相同。  |
33| boolean [outMaxValue](#outmaxvalue) | 若为false则输出索引，为true则输出数值，默认为false。  |
34
35
36## 类成员变量说明
37
38
39### axis
40
41```
42long ArgMaxFusion::axis
43```
44**描述**
45
46指定求最大值索引的维度。
47
48
49### keepDims
50
51```
52boolean ArgMaxFusion::keepDims
53```
54**描述**
55
56是否保持输出维度和输入的维度是否相同。
57
58
59### outMaxValue
60
61```
62boolean ArgMaxFusion::outMaxValue
63```
64**描述**
65
66若为false则输出索引，为true则输出数值，默认为false。
67
68
69### topK
70
71```
72long ArgMaxFusion::topK
73```
74**描述**
75
76轴上前K个最大值。
77