• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# ArgMaxFusion
2
3
4## 概述
5
6返回跨轴的tensor前K个索引或者是数值。
7
8该算子对应的[NodeType](_n_n_rt_v20.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_v20.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### keepDims
49
50```
51boolean ArgMaxFusion::keepDims
52```
53**描述**
54是否保持输出维度和输入的维度是否相同。
55
56
57### outMaxValue
58
59```
60boolean ArgMaxFusion::outMaxValue
61```
62**描述**
63若为false则输出索引,为true则输出数值,默认为false。
64
65
66### topK
67
68```
69long ArgMaxFusion::topK
70```
71**描述**
72轴上前K个最大值。
73