pub trait Module {
// Required method
fn forward(&mut self, input: &Tensor) -> Tensor;
// Provided methods
fn parameters(&self) -> HashMap<String, Arc<RwLock<Tensor>>> { ... }
fn train(&mut self) { ... }
fn eval(&mut self) { ... }
fn zero_grad(&mut self) { ... }
fn visit_parameters(&self, f: &mut dyn FnMut(&str, &Tensor)) { ... }
fn print_parameters(&self, values: bool) { ... }
}Required Methods§
Provided Methods§
fn parameters(&self) -> HashMap<String, Arc<RwLock<Tensor>>>
fn train(&mut self)
fn eval(&mut self)
fn zero_grad(&mut self)
Sourcefn visit_parameters(&self, f: &mut dyn FnMut(&str, &Tensor))
fn visit_parameters(&self, f: &mut dyn FnMut(&str, &Tensor))
Visit all parameters with a callback function
Sourcefn print_parameters(&self, values: bool)
fn print_parameters(&self, values: bool)
Print all parameters and their shapes