|
|
--- |
|
|
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 %} |
|
|
|
