|
| | AddMerge (const bool model=false, const bool run=true) |
| | 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 > &&, 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 > &&, 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 (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 (arma::Mat< eT > &&input, arma::Mat< eT > &&error, arma::Mat< eT > &&gradient) |
| |
| template<typename eT > |
| void | Gradient (arma::Mat< eT > &&input, 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 unsigned int) |
| | Serialize the layer. More...
|
| |
template<typename InputDataType = arma::mat, typename OutputDataType = arma::mat, typename... CustomLayers>
class mlpack::ann::AddMerge< InputDataType, OutputDataType, CustomLayers >
Implementation of the AddMerge module class.
The AddMerge class accumulates the output of various modules.
- Template Parameters
-
| 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.