ArgMaxFusion
概述
返回跨轴的tensor前K个索引或者是数值。
该算子对应的NodeType为NODE_TYPE_ARGMAX_FUSION。
输入：
- x，n维tensor，输入张量（N,*）,其中*意味着数量任意的附加维度。
输出：
- output，轴上输入张量最大值的前K个索引或者是数值。
起始版本： 3.2
相关模块：NNRt
汇总
Public 属性
| 名称 | 描述 |
|---|---|
| long axis | 指定求最大值索引的维度。 |
| long topK | 轴上前K个最大值。 |
| boolean keepDims | 是否保持输出维度和输入的维度是否相同。 |
| boolean outMaxValue | 若为false则输出索引，为true则输出数值，默认为false。 |
类成员变量说明
axis
long ArgMaxFusion::axis
描述 指定求最大值索引的维度。
keepDims
boolean ArgMaxFusion::keepDims
描述 是否保持输出维度和输入的维度是否相同。
outMaxValue
boolean ArgMaxFusion::outMaxValue
描述 若为false则输出索引，为true则输出数值，默认为false。
topK
long ArgMaxFusion::topK
描述 轴上前K个最大值。