GradientSetVisitor Class Reference

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

Inheritance diagram for GradientSetVisitor:

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. More...

Detailed Description

GradientSetVisitor update the gradient parameter given the gradient set.

Definition at line 26 of file gradient_set_visitor.hpp.

Constructor & Destructor Documentation

GradientSetVisitor()

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

Update the gradient parameter given the gradient set.

Member Function Documentation

operator()()

size_t operator()

(

LayerType *

layer

)

const

Update the gradient parameter.

---

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

• src/mlpack/methods/ann/visitor/gradient_set_visitor.hpp