GradientSetVisitor Class Reference
GradientSetVisitor update the gradient parameter given the gradient set. More...
Inheritance diagram for GradientSetVisitor:
Public Member Functions | |
| GradientSetVisitor (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 |
Detailed Description
GradientSetVisitor update the gradient parameter given the gradient set.
Definition at line 26 of file gradient_set_visitor.hpp.
Constructor & Destructor Documentation
◆ GradientSetVisitor()
| GradientSetVisitor | ( | arma::mat & | gradient, |
| size_t | offset = 0 |
||
| ) |
Update the gradient parameter given the gradient set.
Member Function Documentation
◆ operator()() [1/2]
| size_t operator() | ( | LayerType * | layer | ) | const |
Update the gradient parameter.
◆ operator()() [2/2]
| size_t operator() | ( | MoreTypes | layer | ) | const |
The documentation for this class was generated from the following file:
- /home/ryan/src/mlpack.org/_src/mlpack-git/src/mlpack/methods/ann/visitor/gradient_set_visitor.hpp