Method ArgMaxLayer

include/caffe/layers/argmax_layer.hpp:38–39

* @param param provides ArgMaxParameter argmax_param, * with ArgMaxLayer options: * - top_k (\b optional uint, default 1). * the number @f$ K @f$ of maximal items to output. * - out_max_val (\b optional bool, default false). * if set, output a vector of pairs (max_ind, max_val) unless axis is set then * output max_val along the specified axis. * - axis (\

   *     trailing dimensions for each index of the first / num dimension.
   */
  explicit ArgMaxLayer(const LayerParameter& param)
      : Layer<Dtype>(param) {}
  virtual void LayerSetUp(const vector<Blob<Dtype>*>& bottom,
      const vector<Blob<Dtype>*>& top);
  virtual void Reshape(const vector<Blob<Dtype>*>& bottom,