mlpack
|
GradientUpdateVisitor update the gradient parameter given the gradient set. More...
#include <gradient_update_visitor.hpp>
Public Member Functions | |
GradientUpdateVisitor (arma::mat &gradient, size_t offset=0) | |
Update the gradient parameter given the gradient set. More... | |
template<typename LayerType > | |
size_t | operator() (LayerType *layer) const |
Update the gradient parameter. | |
size_t | operator() (MoreTypes layer) const |
GradientUpdateVisitor update the gradient parameter given the gradient set.
|
inline |
Update the gradient parameter given the gradient set.
GradientUpdateVisitor visitor class.