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