Struct NormalizationStats 

pub struct NormalizationStats {
    pub mean: Tensor,
    pub var: Tensor,
    pub running_mean: Option<Tensor>,
    pub running_var: Option<Tensor>,
    pub num_batches_tracked: Option<Tensor>,
}

Normalization statistics for tracking and analysis

Fields

mean: Tensor

var: Tensor

running_mean: Option<Tensor>

running_var: Option<Tensor>

num_batches_tracked: Option<Tensor>

Implementations

impl NormalizationStats

pub fn new(num_features: usize, track_running: bool) -> Result<Self>

Create new normalization statistics

pub fn update_running_stats( &mut self, batch_mean: &Tensor, batch_var: &Tensor, momentum: f32, ) -> Result<()>

Update running statistics

Trait Implementations

impl Clone for NormalizationStats

fn clone(&self) -> NormalizationStats

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for NormalizationStats

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

Formats the value using the given formatter.