Module

ghostflow_nn::module

Trait Module 

Source 
pub trait Module: Send + Sync {
    // Required methods
    fn forward(&self, input: &Tensor) -> Tensor;
    fn parameters(&self) -> Vec<Tensor>;
    fn train(&mut self);
    fn eval(&mut self);
    fn is_training(&self) -> bool;

    // Provided method
    fn num_parameters(&self) -> usize { ... }
}
Expand description

Base trait for all neural network modules

Required Methods§

Source

fn forward(&self, input: &Tensor) -> Tensor

Forward pass

Source

fn parameters(&self) -> Vec<Tensor>

Get all trainable parameters

Source

fn train(&mut self)

Set module to training mode

Source

fn eval(&mut self)

Set module to evaluation mode

Source

fn is_training(&self) -> bool

Check if in training mode

Provided Methods§

Source

fn num_parameters(&self) -> usize

Number of trainable parameters

Implementors§