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### 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