19 lines
700 B
Markdown
19 lines
700 B
Markdown
---
|
|
title: ArgMax Layer
|
|
---
|
|
|
|
# ArgMax Layer
|
|
|
|
* Layer type: `ArgMax`
|
|
* [Doxygen Documentation](http://caffe.berkeleyvision.org/doxygen/classcaffe_1_1ArgMaxLayer.html)
|
|
* Header: [`./include/caffe/layers/argmax_layer.hpp`](https://github.com/BVLC/caffe/blob/master/include/caffe/layers/argmax_layer.hpp)
|
|
* CPU implementation: [`./src/caffe/layers/argmax_layer.cpp`](https://github.com/BVLC/caffe/blob/master/src/caffe/layers/argmax_layer.cpp)
|
|
|
|
## Parameters
|
|
* Parameters (`ArgMaxParameter argmax_param`)
|
|
* From [`./src/caffe/proto/caffe.proto`](https://github.com/BVLC/caffe/blob/master/src/caffe/proto/caffe.proto)):
|
|
|
|
{% highlight Protobuf %}
|
|
{% include proto/ArgMaxParameter.txt %}
|
|
{% endhighlight %}
|