lmp::nets::DropoutImpl Class Reference

Inheritance diagram for lmp::nets::DropoutImpl:

Public Member Functions
	DropoutImpl (tensor::Scalar p)
autograd::Variable	forward (const autograd::Variable &x) const
Public Member Functions inherited from lmp::nets::ModuleImpl
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 ()

Additional Inherited Members
Protected Member Functions inherited from lmp::nets::ModuleImpl
void	register_parameter (const std::string &name, Parameter &param)
void	register_module (const std::string &name, std::shared_ptr< ModuleImpl > module)
Protected Attributes inherited from lmp::nets::ModuleImpl
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/layers/dropout.hpp
• csrc/src/nets/layers/dropout.cpp