|
| MultiplyMerge (const bool model=false, const bool run=true) |
| Create the MultiplyMerge object using the specified parameters. More...
|
|
| ~MultiplyMerge () |
| 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, using the results from the feed forward pass. More...
|
|
OutputDataType & | Delta () |
| Modify the delta. More...
|
|
OutputDataType const & | Delta () const |
| Get 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...
|
|
OutputDataType & | Gradient () |
| Modify the gradient. More...
|
|
OutputDataType const & | Gradient () const |
| Get the gradient. More...
|
|
template<typename eT > |
void | Gradient (const arma::Mat< eT > &input, const arma::Mat< eT > &error, arma::Mat< eT > &gradient) |
|
std::vector< LayerTypes< CustomLayers... > > & | Model () |
| Return the model modules. More...
|
|
OutputDataType & | OutputParameter () |
| Modify the output parameter. More...
|
|
OutputDataType const & | OutputParameter () const |
| Get the output parameter. More...
|
|
OutputDataType & | Parameters () |
| Modify the parameters. More...
|
|
OutputDataType const & | Parameters () const |
| Get the parameters. 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::MultiplyMerge< InputDataType, OutputDataType, CustomLayers >
Implementation of the MultiplyMerge module class.
The MultiplyMerge class multiplies the output of various modules element-wise.
- 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 multiply_merge.hpp.