Caffe: a fast open framework for deep learning.
Add argmax_param "axis" to maximise output along the specified axis
