GradientUpdateVisitor update the gradient parameter given the gradient set. More...
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. More... | |
size_t | operator() (MoreTypes layer) const |
GradientUpdateVisitor update the gradient parameter given the gradient set.
Definition at line 26 of file gradient_update_visitor.hpp.
GradientUpdateVisitor | ( | arma::mat & | gradient, |
size_t | offset = 0 |
||
) |
Update the gradient parameter given the gradient set.
size_t operator() | ( | LayerType * | layer | ) | const |
Update the gradient parameter.
size_t operator() | ( | MoreTypes | layer | ) | const |