ConvNet  1.0
A GPU-based C++ implementation of Convolutional Neural Nets
 All Classes Namespaces Functions Variables
downsample_edge.h
1 #ifndef DOWNSAMPLE_EDGE_H_
2 #define DOWNSAMPLE_EDGE_H_
3 #include "edge.h"
4 
8 class DownSampleEdge : public Edge {
9  public:
10  DownSampleEdge(const config::Edge& edge_config);
11  virtual void ComputeUp(Matrix& input, Matrix& output, bool overwrite);
12  virtual void ComputeDown(Matrix& deriv_output, Matrix& input,
13  Matrix& output, Matrix& deriv_input, bool overwrite);
14  virtual void SetImageSize(int image_size);
15 
16  private:
17  const int sample_factor_;
18 };
19 #endif
virtual void ComputeDown(Matrix &deriv_output, Matrix &input, Matrix &output, Matrix &deriv_input, bool overwrite)
Computes the derivative w.r.t the inputs of this edge given the derivative w.r.t the outputs of this ...
Definition: downsample_edge.cc:21
virtual void SetImageSize(int image_size)
Set the spatial size of the input to this edge.
Definition: downsample_edge.cc:8
This class is intended to be used as a base class for implementing edges.
Definition: edge.h:13
virtual void ComputeUp(Matrix &input, Matrix &output, bool overwrite)
Computes the output layer state given the input.
Definition: downsample_edge.cc:13
A GPU matrix class.
Definition: matrix.h:11
Implements a down-sampling edge.
Definition: downsample_edge.h:8