Argmin

功能

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

输入

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

属性

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

输出

reduced:输出Tensor,数据类型支持int64,数据格式支持ND。

约束与限制

输入data,不支持reduce轴为0的场景。

支持的ONNX版本

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