Struct LayersBuilder 

pub struct LayersBuilder<C> {
    pub n_real_layers: usize,
    pub n_virtual_layers: Option<(usize, Schedule)>,
    pub mamba_block: C,
    pub ignore_first_residual: bool,
    pub ignore_last_residual: bool,
    pub class_latents: Vec<ClassLatent>,
    pub residuals: ResidualsConfig,
}

Plain (non-serde) factory for Layers. The serializable surface is the concrete MambaLatentNetConfig enum; this is just the generic builder it delegates to.

Fields

n_real_layers: usize

Number of real (weight-bearing) layers.

n_virtual_layers: Option<(usize, Schedule)>

Optional virtual-layer scheduling.

mamba_block: C

Shared block config.

ignore_first_residual: bool

Zero the first virtual layer’s residual.

ignore_last_residual: bool

Zero the last virtual layer’s residual.

class_latents: Vec<ClassLatent>

Stack-level class latents (spliced once before the first virtual layer).

residuals: ResidualsConfig

Inter-layer residual scheme (defaults to plain additive).

Implementations

impl<C: MambaBlockConfig> LayersBuilder<C>

pub fn new(n_real_layers: usize, mamba_block: C) -> Self

Builder with no virtual scheduling, no class latents, residuals enabled.

pub fn with_n_virtual_layers(self, n: Option<(usize, Schedule)>) -> Self

Set the optional virtual-layer scheduling.

pub fn with_residuals(self, residuals: ResidualsConfig) -> Self

Set the inter-layer residual scheme (plain additive vs Multi-Gate).

pub fn with_ignore_first_residual(self, ignore: bool) -> Self

Suppress the first virtual layer’s residual (see Layers).

pub fn with_ignore_last_residual(self, ignore: bool) -> Self

Suppress the last virtual layer’s residual (see Layers).

pub fn init(&self, device: &Device) -> Layers<C::Block>

Allocate and initialise the stack on device.