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

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

#include <gradient_set_visitor.hpp>

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

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

Detailed Description

GradientSetVisitor update the gradient parameter given the gradient set.

Constructor & Destructor Documentation

◆ GradientSetVisitor()

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

Update the gradient parameter given the gradient set.

GradientSetVisitor visitor class.


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