Struct ModuleList 

pub struct ModuleList { /* private fields */ }

List of modules with index-based access.

Unlike Sequential, ModuleList doesn’t define a forward pass. It’s useful for holding submodules that need custom control flow.

Example

use aprender::nn::{ModuleList, Linear, Module};

let layers = ModuleList::new()
    .add(Linear::new(10, 10))
    .add(Linear::new(10, 10))
    .add(Linear::new(10, 10));

// Custom forward with residual connections
fn forward(layers: &ModuleList, x: &Tensor) -> Tensor {
    let mut out = x.clone();
    for i in 0..layers.len() {
        let residual = out.clone();
        out = layers.get(i).expect("layer index in bounds").forward(&out);
        out = out.add(&residual);  // residual connection
    }
    out
}

Implementations

impl ModuleList

pub fn new() -> Self

Create an empty ModuleList.

pub fn add<M: Module + 'static>(self, module: M) -> Self

Add a module to the list.

pub fn add_boxed(self, module: Box<dyn Module>) -> Self

Add a boxed module to the list.

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

Get a module by index.

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

Get a mutable module by index.

pub fn len(&self) -> usize

Get the number of modules.

pub fn is_empty(&self) -> bool

Check if the list is empty.

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

Iterate over modules.

impl ModuleList

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

Get all parameters from all modules.

pub fn parameters_mut(&mut self) -> Vec<&mut Tensor>

Get all mutable parameters.

pub fn train(&mut self)

Set all modules to training mode.

pub fn eval(&mut self)

Set all modules to evaluation mode.

Trait Implementations

impl Debug for ModuleList

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ModuleList

fn default() -> Self

Returns the “default value” for a type.