Atlas系列硬件: Atlas 200l DK A2 开发者套件; Atlas 200l DK 开发者套件（型号：3000）; 加速模块、加速卡、智能小站、服务器、集群

异构计算架构: CANN社区版; CANN商用版

AI框架: MindSpore; TensorFlow框架适配（社区版）; TensorFlow框架适配（商用版）; PyTorch框架适配

分布式训练加速套件: MindSpeed

昇腾推理引擎: MindIE

应用使能MindX: MindX DL; MindX Edge; MindX SDK

全流程开发工具: MindStudio

评分

提单

论坛

小AI

ArgMax

输入

x：

是否必填：必填
数据类型：fp32,fp16
参数解释：输入的tensor
规格限制：无

属性

axis：
- 是否必填：非必填
- 数据类型：int
- 参数解释：指定在输入tensor做削减的轴，如果不提供此参数，则每个batch求top_k
- 规格限制：无

out_max_val：

是否必填：非必填
数据类型：bool

参数解释：是否需要输出最大值。

如果out_max_val为True，且有2个TOP：
- 如果有轴axis，只输出每个轴的最大值。
- 如果没有轴axis，则输出最大值的索引和最大值。
```
layer {
  name: "argmax"
  type: "ArgMax"
  bottom: "data"
  top: "indices"
  top: "values"
  argmax_param {
    out_max_val: True
    top_k: 1
  }
```

如果out_max_val为True，且有1个TOP，则输出最大值:

layer {
  name: "argmax"
  type: "ArgMax"
  bottom: "data"
  top: "values"
  argmax_param {
    out_max_val: True
    top_k: 1
    axis: 1
  }
}

如果out_max_val为False，则输出每个轴的最大索引:

layer {
  name: "argmax"
  type: "ArgMax"
  bottom: "data"
  top: "indices"
  argmax_param {
    out_max_val: False
    top_k: 1
    axis: 1
  }
}

规格限制：无

top_k：
- 是否必填：非必填
- 数据类型：int
- 参数解释：默认是1，表示每个axis轴中前top_k个数（取值大于等于1），其取值范围必须在[1,x.shape(axis)]，对应于caffe中的top_k
- 规格限制：当前只支持1

输出

indices：
- 是否必填：非必填
- 数据类型：int32
- 参数解释：输出的最大值的索引
- 规格限制：无

values：
- 是否必填：非必填
- 数据类型：fp32、fp16
- 参数解释：输出的tensor，包含最大值索引或最大值
- 规格限制：无

父主题： 支持Caffe算子清单