Hardmax

功能

计算Hardmax结果，如果元素是指定axis的最大元素则设为1，否则为0。

输入

一个输入：

x：tensor，数据类型：float16、float。

输出

一个输出：

y：一个tensor，和输入x同样的数据类型和shape。

属性

axis：int，默认为-1，含义：指定计算轴。

约束

使用atc工具--precision_mode参数必须为allow_fp32_to_fp16。

支持的ONNX版本

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