文档
注册
评分
提单
论坛
小AI

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词