计算输入Tensor沿指定维度的最大值。如果keepdim等于1,得到的tensor的维数与输入的相同。如果keepdim等于0,那么生成的Tensor会去除被缩减的维度。
data:输入Tensor,数据类型:float16、float、int8、int32、int64、double。
reduced data:输出Tensor,和输入data的数据类型和shape一致。
当输入为float16、float、int32类型时,axes属性可以为空;当输入为其他类型时,需要配置axes属性值。
axes为空时,当前输出不做全维度规约。此时建议修改onnx算子的axes为所有轴;或者在torch导出onnx图前,在torch模型中使用amax来对所有轴规约(例如,x.amax(dim=[0, 1, 2]))。
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18