convnet/docs/fc__edge_8h_source.html

 #ifndef FC_EDGE_H_

 #define FC_EDGE_H_

 #include "edge_with_weight.h"


 class FCEdge : public EdgeWithWeight {

  public:

   FCEdge(const config::Edge& edge_config);

   virtual void AllocateMemory(bool fprop_only);

   virtual void ComputeUp(Matrix& input, Matrix& output, bool overwrite);

   virtual void ComputeDown(Matrix& deriv_output, Matrix& input,

                            Matrix& output, Matrix& deriv_input, bool overwrite);

   virtual void ComputeOuter(Matrix& input, Matrix& deriv_output);


  private:

   void AllocateMemoryBprop();

   void AllocateMemoryFprop();

 };

 #endif

FCEdge::AllocateMemory
virtual void AllocateMemory(bool fprop_only)
Allocate memory for the model.
Definition: fc_edge.cc:26

Matrix
A GPU matrix class.
Definition: matrix.h:11

FCEdge::ComputeOuter
virtual void ComputeOuter(Matrix &input, Matrix &deriv_output)
Computes the gradient for the weights and biases.
Definition: fc_edge.cc:70

FCEdge
Implements a fully-connected edge.
Definition: fc_edge.h:6

FCEdge::ComputeDown
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: fc_edge.cc:55

FCEdge::ComputeUp
virtual void ComputeUp(Matrix &input, Matrix &output, bool overwrite)
Computes the output layer state given the input.
Definition: fc_edge.cc:38

EdgeWithWeight
Base class for all edges which have weights.
Definition: edge_with_weight.h:9