Struct Sequential 

pub struct Sequential { /* private fields */ }

Implementations

impl Sequential

pub fn new(layers: Vec<Box<dyn Module>>) -> Self

pub fn add(&mut self, layer: Box<dyn Module>)

Trait Implementations

impl Module for Sequential

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

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))

Visit all parameters with a callback function

fn print_parameters(&self, values: bool)

Print all parameters and their shapes