Inheritance diagram for lmp::nets::LinearImpl:
Public Member Functions | |
| LinearImpl (size_t in_features, size_t out_features, bool bias=true, tensor::DeviceType device=DEFAULT_DEVICE, tensor::DataType dtype=DEFAULT_DTYPE) | |
| 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 ¶m) |
| 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/linear.hpp
- csrc/src/nets/layers/linear.cpp
