Struct ModuleList 

pub struct ModuleList { /* private fields */ }

A container for holding a list of modules.

Implementations

impl ModuleList

pub fn new() -> Self

Creates a new empty ModuleList.

pub fn from_vec(modules: Vec<Box<dyn Module>>) -> Self

Creates a ModuleList from a vector of modules.

pub fn push<M: Module + 'static>(&mut self, module: M)

Adds a module to the list.

pub fn len(&self) -> usize

Returns the number of modules.

pub fn is_empty(&self) -> bool

Returns true if the list is empty.

pub fn iter(&self) -> impl Iterator<Item = &Box<dyn Module>>

Returns an iterator over the modules.

pub fn iter_mut(&mut self) -> impl Iterator<Item = &mut Box<dyn Module>>

Returns a mutable iterator over the modules.

pub fn get(&self, index: usize) -> Option<&dyn Module>

Gets a module by index.

Trait Implementations

impl Default for ModuleList

fn default() -> Self

Returns the “default value” for a type.

impl Module for ModuleList

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

Performs the forward pass.

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

Returns all parameters of this module.

fn named_parameters(&self) -> HashMap<String, Parameter>

Returns named parameters of this module.

fn set_training(&mut self, training: bool)

Sets the training mode.

fn is_training(&self) -> bool

Returns whether the module is in training mode.

fn name(&self) -> &'static str

Returns the module name for debugging.

fn num_parameters(&self) -> usize

Returns the number of trainable parameters.

fn train(&mut self)

Sets the module to training mode.

fn eval(&mut self)

Sets the module to evaluation mode.

fn zero_grad(&self)

Zeros all gradients of parameters.