mlpack
Public Member Functions | List of all members
mlpack::ann::GradientZeroVisitor Class Reference
Inheritance diagram for mlpack::ann::GradientZeroVisitor:
Inheritance graph
[legend]
Collaboration diagram for mlpack::ann::GradientZeroVisitor:
Collaboration graph
[legend]

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.
 
void operator() (MoreTypes layer) const
 

Constructor & Destructor Documentation

◆ GradientZeroVisitor()

mlpack::ann::GradientZeroVisitor::GradientZeroVisitor ( )
inline

Set the gradient to zero for the given module.

GradientZeroVisitor visitor class.


The documentation for this class was generated from the following files: