---
title: ArgMax
description: "返回输入Tensor指定轴上最大值对应的索引。"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/aolapi/atlasoxol_09_0010.html
sourcePath: /source/zh/canncommercial/900/API/aolapi/atlasoxol_09_0010.html
indexId: 56c04abdbcea630069e4e59bc0f99bbcdaf430709480c1c6b9665d40853fc06f63
---
# ArgMax

#### 功能

返回输入Tensor指定轴上最大值对应的索引。


#### 输入

data：输入Tensor，数据类型支持float16、float、int32，数据格式支持ND。


#### 属性

- axis：表示计算最大值索引的方向，数据类型：int32，axis的值为[-len(x.shape), len(x.shape)-1]。
- keepdims：keepdims默认为1，支持1或0（可选）。
- select_last_index：如果最大值出现在多个位置，是否选择第一个位置或最后一个位置，取值范围：[0, 1]，默认为0，目前只支持0。


#### 输出

reduced：输出Tensor，表示最大值的索引位置，维度比输入data少1，数据类型支持int64，数据格式支持ND。


#### 约束与限制

无


#### 支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18