- 如果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
}
}