namespace caffe {
-/*
-ConcatLayer
- Takes at least two blobs and concatenates them along either num or
- channel dim, outputting the result.
+/* ArgmaxLayer
+ Compute the index of the max value across all (channels x height x width).
+ [In the future, can take specific dimension.]
+ Intended for use after a classification layer to produce prediction.
+ If parameter out_max_val is set to true, then output is a vector of pairs
+ (max_ind, max_val) for each image.
+
+ NOTE: does not implement Backwards operation.
*/
template <typename Dtype>
class ArgMaxLayer : public Layer<Dtype> {
protected:
virtual Dtype Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
- // For now ArgMax layer should not be used to compute backward operations.
virtual void Backward_cpu(const vector<Blob<Dtype>*>& top,
const bool propagate_down, vector<Blob<Dtype>*>* bottom) {
NOT_IMPLEMENTED;
bool out_max_val_;
};
+/* ConcatLayer
+ Takes at least two blobs and concatenates them along either num or
+ channel dim, outputting the result.
+*/
template <typename Dtype>
class ConcatLayer : public Layer<Dtype> {
public:
const bool propagate_down, vector<Blob<Dtype>*>* bottom);
};
+/* FlattenLayer
+*/
template <typename Dtype>
class FlattenLayer : public Layer<Dtype> {
public:
int pos_;
};
+/* PoolingLayer
+*/
template <typename Dtype>
class PoolingLayer : public Layer<Dtype> {
public: