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.