mlpack 3.4.2
Public Member Functions | List of all members
GradientZeroVisitor Class Reference

#include <gradient_zero_visitor.hpp>

Inheritance diagram for GradientZeroVisitor:
Inheritance 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. 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: