Implementation of the AddMerge module class. More...
Public Member Functions | |
AddMerge (const bool model=false, const bool run=true) | |
Create the AddMerge object using the specified parameters. More... | |
AddMerge (const bool model, const bool run, const bool ownsLayers) | |
Create the AddMerge object using the specified parameters. More... | |
~AddMerge () | |
Destructor to release allocated memory. More... | |
template<class LayerType , class... Args> | |
void | Add (Args... args) |
void | Add (LayerTypes< CustomLayers... > layer) |
template < typename eT > | |
void | Backward (const arma::Mat< eT > &, const arma::Mat< eT > &gy, arma::Mat< eT > &g) |
Ordinary feed backward pass of a neural network, calculating the function f(x) by propagating x backwards trough f. More... | |
template < typename eT > | |
void | Backward (const arma::Mat< eT > &, const arma::Mat< eT > &gy, arma::Mat< eT > &g, const size_t index) |
This is the overload of Backward() that runs only a specific layer with the given input. More... | |
OutputDataType const & | Delta () const |
Get the delta. More... | |
OutputDataType & | Delta () |
Modify the delta. More... | |
template < typename InputType , typename OutputType > | |
void | Forward (const InputType &, OutputType &output) |
Ordinary feed forward pass of a neural network, evaluating the function f(x) by propagating the activity forward through f. More... | |
template < typename eT > | |
void | Gradient (const arma::Mat< eT > &input, const arma::Mat< eT > &error, arma::Mat< eT > &gradient) |
template < typename eT > | |
void | Gradient (const arma::Mat< eT > &input, const arma::Mat< eT > &error, arma::Mat< eT > &gradient, const size_t index) |
InputDataType const & | InputParameter () const |
Get the input parameter. More... | |
InputDataType & | InputParameter () |
Modify the input parameter. More... | |
std::vector< LayerTypes< CustomLayers... > > & | Model () |
Return the model modules. More... | |
OutputDataType const & | OutputParameter () const |
Get the output parameter. More... | |
OutputDataType & | OutputParameter () |
Modify the output parameter. More... | |
OutputDataType const & | Parameters () const |
Get the parameters. More... | |
OutputDataType & | Parameters () |
Modify the parameters. More... | |
bool | Run () const |
Get the value of run parameter. More... | |
bool & | Run () |
Modify the value of run parameter. More... | |
template < typename Archive > | |
void | serialize (Archive &ar, const uint32_t) |
Serialize the layer. More... | |
Implementation of the AddMerge module class.
The AddMerge class accumulates the output of various modules.
InputDataType | Type of the input data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
OutputDataType | Type of the output data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
CustomLayers | Additional custom layers that can be added. |
Definition at line 42 of file add_merge.hpp.
AddMerge | ( | const bool | model = false , |
const bool | run = true |
||
) |
Create the AddMerge object using the specified parameters.
model | Expose all the network modules. |
run | Call the Forward/Backward method before the output is merged. |
AddMerge | ( | const bool | model, |
const bool | run, | ||
const bool | ownsLayers | ||
) |
Create the AddMerge object using the specified parameters.
model | Expose all the network modules. |
run | Call the Forward/Backward method before the output is merged. |
ownsLayers | Delete the layers when this is deallocated. |
~AddMerge | ( | ) |
Destructor to release allocated memory.
|
inline |
Definition at line 137 of file add_merge.hpp.
|
inline |
Definition at line 144 of file add_merge.hpp.
void Backward | ( | const arma::Mat< eT > & | , |
const arma::Mat< eT > & | gy, | ||
arma::Mat< eT > & | g | ||
) |
Ordinary feed backward pass of a neural network, calculating the function f(x) by propagating x backwards trough f.
Using the results from the feed forward pass.
* | (input) The propagated input activation. |
gy | The backpropagated error. |
g | The calculated gradient. |
void Backward | ( | const arma::Mat< eT > & | , |
const arma::Mat< eT > & | gy, | ||
arma::Mat< eT > & | g, | ||
const size_t | index | ||
) |
This is the overload of Backward() that runs only a specific layer with the given input.
* | (input) The propagated input activation. |
gy | The backpropagated error. |
g | The calculated gradient. |
index | The index of the layer to run. |
|
inline |
Get the delta.
Definition at line 157 of file add_merge.hpp.
|
inline |
Modify the delta.
Definition at line 159 of file add_merge.hpp.
void Forward | ( | const InputType & | , |
OutputType & | output | ||
) |
Ordinary feed forward pass of a neural network, evaluating the function f(x) by propagating the activity forward through f.
* | (input) Input data used for evaluating the specified function. |
output | Resulting output activation. |
void Gradient | ( | const arma::Mat< eT > & | input, |
const arma::Mat< eT > & | error, | ||
arma::Mat< eT > & | gradient | ||
) |
void Gradient | ( | const arma::Mat< eT > & | input, |
const arma::Mat< eT > & | error, | ||
arma::Mat< eT > & | gradient, | ||
const size_t | index | ||
) |
|
inline |
Get the input parameter.
Definition at line 147 of file add_merge.hpp.
|
inline |
Modify the input parameter.
Definition at line 149 of file add_merge.hpp.
|
inline |
Return the model modules.
Definition at line 162 of file add_merge.hpp.
|
inline |
Get the output parameter.
Definition at line 152 of file add_merge.hpp.
|
inline |
Modify the output parameter.
Definition at line 154 of file add_merge.hpp.
|
inline |
Get the parameters.
Definition at line 173 of file add_merge.hpp.
|
inline |
Modify the parameters.
Definition at line 175 of file add_merge.hpp.
|
inline |
Get the value of run parameter.
Definition at line 178 of file add_merge.hpp.
|
inline |
Modify the value of run parameter.
Definition at line 180 of file add_merge.hpp.
References AddMerge< InputDataType, OutputDataType, CustomLayers >::serialize().
void serialize | ( | Archive & | ar, |
const uint32_t | |||
) |
Serialize the layer.
Referenced by AddMerge< InputDataType, OutputDataType, CustomLayers >::Run().