Argmax

Description

Returns the index corresponding to the maximum value on the specified axis.

Input

One input:

x: tensor of type float16 or float.

Output

One output:

y: tensor of type int64, indicating the index position of the maximum value. The dimension is one less than that of the input x.

Property

axis: (required) int32, specifying the direction for calculating the maximum index. The value for aixs is [-len(x.shape), len(x.shape)-1].

keep_dim: (optional) either 1 (default) or 0.

ONNX Opset Support

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