#include <gradtable.h>
Public Member Functions | |
GradTable () | |
unsigned int | get_size () |
Tensor< T > * | get (Operation< T > *var) |
void | set (Operation< T > *var, Tensor< T > *grad) |
void | clear () |
Protected Attributes | |
std::map< Operation< T > *, Tensor< T > *> | _table |
std::map< Operation< T > *, Tensor< T > * >::iterator | tmp_map_iterator |
GradTable class.
T | Numeric |
magmadnn::op::GradTable< T >::GradTable | ( | ) |
Constructs a new grad table.
void magmadnn::op::GradTable< T >::clear | ( | ) |
Removes all entries.
Tensor< T > * magmadnn::op::GradTable< T >::get | ( | Operation< T > * | var | ) |
Takes a variable and returns the tree to compute its gradient.
var |
unsigned int magmadnn::op::GradTable< T >::get_size | ( | ) |
returns the size of this grad table.
void magmadnn::op::GradTable< T >::set | ( | Operation< T > * | var, |
Tensor< T > * | grad | ||
) |
Sets var's gradient to grad.
var | |
grad |