- Ascend 950PR/Ascend 950DT：支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持
对一个多维向量在指定的维度求最大值的索引。
定义指定计算的维度（Reduce轴）为R轴，非指定维度（Normal轴）为A轴。对Shape为(2, 3)的二维矩阵进行运算，指定在第一维求最大值索引，输出结果为[1, 1, 1]；指定在第二维求最大值索引，输出结果为[2, 2]。
[object Object]
返回输出Tensor，输出Tensor的数据类型为DT_INT32，Shape与keepdim参数相关。
若keepdim参数为True，则在执行归约操作后保留被归约的维度。输出Tensor在除dim指定的维度外，其他维度的Shape与输入Tensor的Shape一致，而在dim指定的维度上的大小为1。
若keepdim参数为False（默认），则被归约的维度会从输出Tensor中移除，而tileshape中对应的维度不变,所以建议在调其他operation前重设tileshape。
TileShape大小不超过64KB；
尾轴要32bytes对齐；
Tensor类型输入不支持
[object Object]格式。
调用该operation接口前，应通过set_vec_tile_shapes设置TileShape。
TileShape维度应和输入input一致。
如输入input shape为[m, n]，输出为[m, 1]，TileShape设置为[m1, n1],则m1, n1分别用于切分m, n轴。
[object Object]
注意：如果keepdim设置为false，则被归约的维度会从输出Tensor中移除，而tileshape中对应的维度不变,所以建议在调其他operation前重设tileshape。
[object Object]
结果示例如下：
[object Object]