mlpack
Public Member Functions | List of all members
mlpack::ann::GradientUpdateVisitor Class Reference

GradientUpdateVisitor update the gradient parameter given the gradient set. More...

#include <gradient_update_visitor.hpp>

Inheritance diagram for mlpack::ann::GradientUpdateVisitor:
Inheritance graph
[legend]
Collaboration diagram for mlpack::ann::GradientUpdateVisitor:
Collaboration graph
[legend]

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

Detailed Description

GradientUpdateVisitor update the gradient parameter given the gradient set.

Constructor & Destructor Documentation

◆ GradientUpdateVisitor()

mlpack::ann::GradientUpdateVisitor::GradientUpdateVisitor ( arma::mat &  gradient,
size_t  offset = 0 
)
inline

Update the gradient parameter given the gradient set.

GradientUpdateVisitor visitor class.


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