ceras::gradient_descent< Loss, T > Struct Template Reference

#include <optimizer.hpp>

Inheritance diagram for ceras::gradient_descent< Loss, T >:

Public Types
typedef tensor< T >	tensor_type

Public Member Functions
	gradient_descent (Loss &loss, std::size_t batch_size, T learning_rate=1.0e-3, T momentum=0.0) noexcept
void	forward ()

Public Attributes
Loss &	loss_
T	learning_rate_
T	momentum_

Member Typedef Documentation

tensor_type

template<typename Loss , typename T >

typedef tensor< T > ceras::gradient_descent< Loss, T >::tensor_type

Constructor & Destructor Documentation

gradient_descent()

template<typename Loss , typename T >

ceras::gradient_descent< Loss, T >::gradient_descent ( Loss &  loss, std::size_t  batch_size, T  learning_rate = 1.0e-3, T  momentum = 0.0  )

inlinenoexcept

Member Function Documentation

forward()

template<typename Loss , typename T >

void ceras::gradient_descent< Loss, T >::forward ( )

inline

Member Data Documentation

learning_rate_

template<typename Loss , typename T >

T ceras::gradient_descent< Loss, T >::learning_rate_

loss_

template<typename Loss , typename T >

Loss& ceras::gradient_descent< Loss, T >::loss_

momentum_

template<typename Loss , typename T >

T ceras::gradient_descent< Loss, T >::momentum_

---

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

• /home/feng/workspace/github.repo/ceras/include/optimizer.hpp