constant.hpp
Go to the documentation of this file.
1 
13 #ifndef MLPACK_METHODS_ANN_LAYER_CONSTANT_HPP
14 #define MLPACK_METHODS_ANN_LAYER_CONSTANT_HPP
15 
16 #include <mlpack/prereqs.hpp>
17 
18 namespace mlpack {
19 namespace ann {
20 
30 template <
31  typename InputDataType = arma::mat,
32  typename OutputDataType = arma::mat
33 >
34 class Constant
35 {
36  public:
44  Constant(const size_t outSize = 0, const double scalar = 0.0);
45 
53  template<typename InputType, typename OutputType>
54  void Forward(const InputType&& input, OutputType&& output);
55 
64  template<typename DataType>
65  void Backward(const DataType&& /* input */,
66  DataType&& /* gy */,
67  DataType&& g);
68 
70  OutputDataType& OutputParameter() const { return outputParameter; }
72  OutputDataType& OutputParameter() { return outputParameter; }
73 
75  OutputDataType& Delta() const { return delta; }
77  OutputDataType& Delta() { return delta; }
78 
82  template<typename Archive>
83  void serialize(Archive& ar, const unsigned int /* version */);
84 
85  private:
87  size_t inSize;
88 
90  size_t outSize;
91 
93  OutputDataType constantOutput;
94 
96  OutputDataType delta;
97 
99  OutputDataType outputParameter;
100 }; // class ConstantLayer
101 
102 } // namespace ann
103 } // namespace mlpack
104 
105 // Include implementation.
106 #include "constant_impl.hpp"
107 
108 #endif
OutputDataType & OutputParameter()
Modify the output parameter.
Definition: constant.hpp:72
.hpp
Definition: add_to_po.hpp:21
Constant(const size_t outSize=0, const double scalar=0.0)
Create the Constant object that outputs a given constant scalar value given any input value...
The core includes that mlpack expects; standard C++ includes and Armadillo.
OutputDataType & OutputParameter() const
Get the output parameter.
Definition: constant.hpp:70
void Forward(const InputType &&input, OutputType &&output)
Ordinary feed forward pass of a neural network.
OutputDataType & Delta()
Modify the delta.
Definition: constant.hpp:77
void Backward(const DataType &&, DataType &&, DataType &&g)
Ordinary feed backward pass of a neural network.
OutputDataType & Delta() const
Get the delta.
Definition: constant.hpp:75
Implementation of the constant layer.
Definition: constant.hpp:34
void serialize(Archive &ar, const unsigned int)
Serialize the layer.