Struct DiagonalGaussian 

pub struct DiagonalGaussian {
    pub mean: Vec<f32>,
    pub log_var: Vec<f32>,
}

Diagonal Gaussian parameters

Fields

mean: Vec<f32>

Mean vector

log_var: Vec<f32>

Log variance vector

Implementations

impl DiagonalGaussian

pub fn new(mean: Vec<f32>, log_var: Vec<f32>) -> Self

Create from mean and log variance

pub fn unit(dim: usize) -> Self

Create unit Gaussian (mean=0, var=1)

pub fn sample(&self, epsilon: &[f32]) -> Vec<f32>

Sample using reparameterization trick z = mean + std * epsilon, where epsilon ~ N(0, 1)

pub fn variance(&self) -> Vec<f32>

Get variance

pub fn std(&self) -> Vec<f32>

Get standard deviation

Trait Implementations

impl Clone for DiagonalGaussian

fn clone(&self) -> DiagonalGaussian

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DiagonalGaussian

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

Formats the value using the given formatter.