hardmax

功能

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

边界

【输入】

1个输入

x：tensor，rank=2，数据类型：float16、float32

【输出】

一个输出

y：一个张量，和输入x同样的type和shape

【属性】

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

【约束】

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

支持的ONNX版本

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