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
父主题: 支持ONNX算子清单