pub trait Module {
type Config: Config;
type Params;
// Required methods
fn config(&self) -> &Self::Config;
fn params(&self) -> &Self::Params;
fn params_mut(&mut self) -> &mut Self::Params;
}Expand description
A Module defines any object that may be used as a layer in a neural network.
Config is a type that defines the configuration of the module; including any and all hyperparameters.
Params is a type that defines the parameters of the module; typically references a Linear set of parameters { weights, bias }