Argmin
Description
Returns the index corresponding to the minimum value on the specified axis of the input tensor.
Input
One input:
x: tensor of type float16 or float.
Output
One output:
y: tensor of type int64.
Property
axis: int, specifying the calculation axis. Value range: [-r, r-1], where r indicates the rank of the input data.
ONNX Opset Support
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
Parent topic: Operators Specifications of the AI Framework