BVLC/caffe — pull request #421 by sguada (2014-05-16 16:50:15+00:00)
Title: Argmax layer

Currently the layer compute the index of the maximum value for each image across all features. It is intended to be used after a classification layer to produce the predicted label.
If parameter`out_max_val` is set to true then it will produce a vector of pairs (max_ind,max_val) for each image.
In the future it could accept specific dimensions to compute the maximum and argmax.