Struct NormalizationConfig 

pub struct NormalizationConfig {
    pub eps: f32,
    pub momentum: f32,
    pub affine: bool,
    pub track_running_stats: bool,
}

Configuration for normalization layers

Fields

eps: f32

Small constant added to variance for numerical stability

momentum: f32

Momentum for running statistics update

affine: bool

Whether to use learnable affine parameters (weight and bias)

track_running_stats: bool

Whether to track running statistics for batch norm

Implementations

impl NormalizationConfig

pub fn training() -> Self

Create configuration for training mode with tracking

pub fn inference() -> Self

Create configuration for inference mode without tracking

pub fn non_affine() -> Self

Create configuration without learnable parameters

pub fn with_eps(eps: f32) -> Self

Create configuration with custom epsilon for numerical stability

pub fn with_momentum(momentum: f32) -> Self

Create configuration with custom momentum for running stats

Trait Implementations

impl Clone for NormalizationConfig

fn clone(&self) -> NormalizationConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for NormalizationConfig

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

Formats the value using the given formatter.

impl Default for NormalizationConfig

fn default() -> Self

Returns the “default value” for a type.