GradientZeroVisitor Class Reference
Inheritance diagram for GradientZeroVisitor:
Public Member Functions | |
| GradientZeroVisitor () | |
| Set the gradient to zero for the given module. More... | |
template < typename LayerType > | |
| void | operator() (LayerType *layer) const |
| Set the gradient to zero. More... | |
| void | operator() (MoreTypes layer) const |
Detailed Description
Definition at line 27 of file gradient_zero_visitor.hpp.
Constructor & Destructor Documentation
◆ GradientZeroVisitor()
Set the gradient to zero for the given module.
Member Function Documentation
◆ operator()() [1/2]
| void operator() | ( | LayerType * | layer | ) | const |
Set the gradient to zero.
◆ operator()() [2/2]
| void 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_zero_visitor.hpp