lmp::nets::ModuleImpl Class Reference

Inheritance diagram for lmp::nets::ModuleImpl:

Public Member Functions
std::vector< Parameter >	parameters () const
std::multimap< std::string, Parameter >	named_parameters () const
std::vector< std::reference_wrapper< Parameter > >	parameters ()
std::multimap< std::string, std::reference_wrapper< Parameter > >	named_parameters ()
void	eval ()
void	train ()

Protected Member Functions
void	register_parameter (const std::string &name, Parameter &param)
void	register_module (const std::string &name, std::shared_ptr< ModuleImpl > module)

Protected Attributes
bool	trainable_ = true
std::unordered_map< std::string, std::shared_ptr< ModuleImpl > >	modules_
std::unordered_map< std::string, std::reference_wrapper< Parameter > >	params_

---

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

• csrc/include/lamppp/nets/module.hpp
• csrc/src/nets/module.cpp