Struct NormalizingFlowChain 

pub struct NormalizingFlowChain {
    pub layers: Vec<FlowLayer>,
}

A chain of normalizing flow layers

Fields

layers: Vec<FlowLayer>

Ordered list of flow layers

Implementations

impl NormalizingFlowChain

pub fn new(layers: Vec<FlowLayer>) -> Self

Create a new flow chain with the given layers

pub fn planar(dim: usize, n_layers: usize, seed: u64) -> Self

Create a chain of planar flows

pub fn radial(dim: usize, n_layers: usize, seed: u64) -> Self

Create a chain of radial flows

pub fn mixed(dim: usize, n_layers: usize, seed: u64) -> Self

Create a mixed chain alternating planar and radial flows

pub fn forward(&self, z0: &Array1<f64>) -> StatsResult<(Array1<f64>, f64)>

Apply the full chain: z_K = f_K . … . f_1(z_0)

Returns (z_K, sum of log|det J_k|)

pub fn n_params(&self) -> usize

Total number of flow parameters across all layers

pub fn get_params(&self) -> Array1<f64>

Get all flow parameters as a flat vector

pub fn set_params(&mut self, params: &Array1<f64>) -> StatsResult<()>

Set all flow parameters from a flat vector

Trait Implementations

impl Clone for NormalizingFlowChain

fn clone(&self) -> NormalizingFlowChain

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for NormalizingFlowChain

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

Formats the value using the given formatter.