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