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

 
size_t operator() (MoreTypes layer) const
 

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()() [1/2]

size_t operator() ( LayerType *  layer) const

Update the gradient parameter.

◆ operator()() [2/2]

size_t operator() ( MoreTypes  layer) const

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