ArgMax

功能

返回输入Tensor指定轴上最大值对应的索引。

输入

data：输入Tensor，数据类型支持float16、float、int32，数据格式支持ND。

属性

• axis：表示计算最大值索引的方向，数据类型：int32，axis的值为[-len(x.shape), len(x.shape)-1]。
• keepdims：keepdims默认为1，支持1或0（可选）。
• select_last_index：如果最大值出现在多个位置，是否选择第一个位置或最后一个位置，取值范围：[0, 1]，默认为0，目前只支持0。

输出

reduced：输出Tensor，表示最大值的索引位置，维度比输入x少1，数据类型支持int64，数据格式支持ND。

约束与限制

无

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18