Argmax

功能

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

输入

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

属性

• axis：表示计算最大值索引的方向，数据类型：int32，aixs的值为[-len(x.shape), len(x.shape)-1]。
• keep_dim：keep_dim默认为1，支持1或0（可选）。

输出

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

约束与限制

无

支持的ONNX版本

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