ArgMin

功能

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

输入

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

属性

• axis：数据类型为int，含义：指定计算轴；取值范围：[-r, r-1]，r表示输入数据的秩。
• keepdims：keepdims默认为1，支持1或0（可选）。
• select_last_index：如果最小值出现在多个位置，是否选择第一个位置或最后一个位置，取值范围：[0, 1]，默认为0，目前只支持0。

输出

reduced：输出Tensor，数据类型支持int32，数据格式支持ND。

约束与限制

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

支持的ONNX版本

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