Argmax

功能

返回指定轴上最大值所对应的索引。

输入

一个输入：

x：一个tensor，数据类型：float16、float。

输出

一个输出：

y：一个tensor，表示最大值的索引位置，维度比输入x少1，数据类型：int64。

属性

axis：必选，表示计算最大值索引的方向，数据类型：int32，aixs的值为[-len(x.shape), len(x.shape)-1]。

keep_dim：可选，keep_dim默认为1，支持1或0。

支持的ONNX版本

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