GradientUpdateVisitor Class Reference

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

Inheritance diagram for GradientUpdateVisitor:

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

 

Detailed Description

GradientUpdateVisitor update the gradient parameter given the gradient set.

Definition at line 26 of file gradient_update_visitor.hpp.

Constructor & Destructor Documentation

◆ GradientUpdateVisitor()

GradientUpdateVisitor ( 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: