Argmin

功能

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

输入

一个输入：

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

输出

一个输出：

y：一个tensor，数据类型：int64。

属性

axis：数据类型为int，含义：指定计算轴；取值范围：[-r, r-1]，r表示输入数据的秩。

支持的ONNX版本

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