ArgMax

输入

x：

是否必填：必填
数据类型：float32、float16
参数解释：输入的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：
- 是否必填：非必填
- 数据类型：float32、float16
- 参数解释：输出的tensor，包含最大值索引或最大值
- 规格限制：无

父主题： 支持Caffe算子清单