ArgMax.md 636 字节
Newer Older
S
sunyanfang01 已提交
1 2 3 4 5 6 7 8 9 10
## ArgMax


### [ArgMax](http://caffe.berkeleyvision.org/tutorial/layers/argmax.html)
```
layer {
    name: "argmax"
    type: "ArgMax"
    bottom: "data"
    top: "argmax"	
S
SunAhong1993 已提交
11
    argmax_param {
S
sunyanfang01 已提交
12 13 14 15 16 17 18 19
	out_max_val: false
	top_k: 1
	axis: 0
    }
}
```


S
SunAhong1993 已提交
20
### [paddle.fluid.layers.argmax](http://paddlepaddle.org/documentation/docs/zh/1.4/api_cn/layers_cn.html#permalink-214-argmax)
S
sunyanfang01 已提交
21 22 23
```python
paddle.fluid.layers.argmax(
    x,
S
SunAhong1993 已提交
24
    axis=0
S
sunyanfang01 已提交
25 26 27 28
)
```  

### 功能差异
J
Jason 已提交
29
#### 计算机制
J
Jason 已提交
30
Caffe:可通过`top_k``out_max_val`参数设置得到前`k`的索引或数值;                            
J
Jason 已提交
31
PaddlePaddle:只能输出最大值的索引;