返回指定轴上最大值所对应的索引
【输入】
一个输入
x：一个tensor，数据类型：float16、float32
【输出】
一个输出
y：一个张量，表示最大值的索引位置，维度比输入x少1，数据类型：int32
【属性】
axis：必选，表示计算最大值索引的方向，数据类型：int32，aixs的值为[-len(x.shape), len(x.shape)-1]
keep_dim：可选，keep_dim默认为1，支持1或0。
【约束】
算子不支持atc工具参数--precision_mode=must_keep_origin_dtype时float32类型输入
Opset v8/v9/v10/v11/v12/v13