Hardmax

功能

根据axis找到指定轴,沿指定轴找出第一个最大值,该轴中第一个最大值位置设置为1,其余位置设置为0。

输入

x:输入Tensor,输入Tensor维度>=axis,数据类型:float16、float。

属性

axis:数据类型支持int,默认为-1,含义:表示Hardmax沿哪个维度将执行。

输出

y:输出Tensor,和输入x同样的数据类型和shape。

约束

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

支持的ONNX版本

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