根据axis找到指定轴,沿指定轴找出第一个最大值,该轴中第一个最大值位置设置为1,其余位置设置为0。
x:输入Tensor,输入Tensor维度>=axis,数据类型:float16、float。
axis:数据类型支持int,默认为-1,含义:表示Hardmax沿哪个维度将执行。
y:输出Tensor,和输入x同样的数据类型和shape。
使用atc工具--precision_mode参数必须为allow_fp32_to_fp16。
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18