Inheritance diagram for magmadnn::optimizer::GradientDescent< T >:

Collaboration diagram for magmadnn::optimizer::GradientDescent< T >:

Public Member Functions
	GradientDescent (T learning_rate)

virtual void	minimize (op::Operation< T > obj_func, const std::vector< op::Operation< T > > &wrt)

void	set_learning_rate (T learning_rate)

T	get_learning_rate ()

Public Member Functions inherited from magmadnn::optimizer::Optimizer< T >
virtual std::string	get_name ()

Protected Member Functions
virtual void	update (op::Operation< T > var, Tensor< T > grad)

Protected Attributes
T	learning_rate

op::GradTable< T >	table

Protected Attributes inherited from magmadnn::optimizer::Optimizer< T >
op::Operation< T > *	_obj_func

std::string	_name = "Generic Optimizer"

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

Public Member Functions

Protected Member Functions