SearchModeVisitor executes the Gradient() method of the given module using the input and delta parameter.
More...
|
| GradientVisitor (arma::mat &&input, arma::mat &&delta) |
| Executes the Gradient() method of the given module using the input and delta parameter. More...
|
|
template<typename LayerType > |
void | operator() (LayerType *layer) const |
| Executes the Gradient() method. More...
|
|
|
template<typename T > |
std::enable_if< HasGradientCheck< T, arma::mat &(T::*)()>::value, void >::type | LayerGradients (T *layer, arma::mat &input) const |
| Execute the Gradient() function if the module implements the Gradient() function. More...
|
|
template<typename T , typename P > |
std::enable_if< !HasGradientCheck< T, P &(T::*)()>::value, void >::type | LayerGradients (T *layer, P &input) const |
| Do not execute the Gradient() function if the module doesn't implement the Gradient() function. More...
|
|
SearchModeVisitor executes the Gradient() method of the given module using the input and delta parameter.
Definition at line 28 of file gradient_visitor.hpp.
mlpack::ann::GradientVisitor::GradientVisitor |
( |
arma::mat && |
input, |
|
|
arma::mat && |
delta |
|
) |
| |
Executes the Gradient() method of the given module using the input and delta parameter.
template<typename T >
std::enable_if< HasGradientCheck<T, arma::mat&(T::*)()>::value, void>::type mlpack::ann::GradientVisitor::LayerGradients |
( |
T * |
layer, |
|
|
arma::mat & |
input |
|
) |
| const |
|
private |
Execute the Gradient() function if the module implements the Gradient() function.
template<typename T , typename P >
std::enable_if< !HasGradientCheck<T, P&(T::*)()>::value, void>::type mlpack::ann::GradientVisitor::LayerGradients |
( |
T * |
layer, |
|
|
P & |
input |
|
) |
| const |
|
private |
Do not execute the Gradient() function if the module doesn't implement the Gradient() function.
template<typename LayerType >
void mlpack::ann::GradientVisitor::operator() |
( |
LayerType * |
layer | ) |
const |
Executes the Gradient() method.
arma::mat&& mlpack::ann::GradientVisitor::delta |
|
private |
arma::mat&& mlpack::ann::GradientVisitor::input |
|
private |
The documentation for this class was generated from the following file: