Argmin

功能

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

边界

【输入】

一个输入

x:一个tensor,数据类型:float16、float32

【输出】

一个输出

y:一个tensor,数据类型:int64

【属性】

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

【约束】

算子不支持atc工具参数--precision_mode=must_keep_origin_dtype时float32类型输入

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13