Enum FlowLayer 

pub enum FlowLayer {
    Planar {
        w: Array1<f64>,
        u: Array1<f64>,
        b: f64,
    },
    Radial {
        z0: Array1<f64>,
        alpha: f64,
        beta: f64,
    },
}

Type of normalizing flow layer

Variants

Planar

Planar flow: f(z) = z + u * tanh(w^T z + b)

Fields

w: Array1<f64>

Weight vector w (dim)

u: Array1<f64>

Scale vector u (dim)

b: f64

Bias scalar

Radial

Radial flow: f(z) = z + beta * (z - z0) / (alpha + ||z - z0||)

Fields

z0: Array1<f64>

Center point z0 (dim)

alpha: f64

Scale parameter alpha > 0

beta: f64

Magnitude parameter beta

Implementations

impl FlowLayer

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

Create a new planar flow layer with random initialization

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

Create a new radial flow layer with random initialization

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

Apply the flow transformation: f(z) and compute log|det(df/dz)|

Returns (f(z), log|det J|)

pub fn n_params(&self) -> usize

Get the total number of parameters for this flow layer

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

Get all parameters as a flat vector

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

Set parameters from a flat vector

Trait Implementations

impl Clone for FlowLayer

fn clone(&self) -> FlowLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FlowLayer

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

Formats the value using the given formatter.