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/>指定求最大值索引的维度。 | 35| [topK](#topk) | long<br/>轴上前K个最大值。 | 36| [keepDims](#keepdims) | boolean<br/>是否保持输出维度和输入的维度是否相同。 | 37| [outMaxValue](#outmaxvalue) | boolean<br/>若为false则输出索引,为true则输出数值,默认为false。 | 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