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