Struct BatchNorm1d 

pub struct BatchNorm1d { /* private fields */ }

Batch Normalization for 1D inputs (N, C) or (N, C, L)

Implementations

impl BatchNorm1d

pub fn new(num_features: usize) -> Self

pub fn with_params(num_features: usize, eps: f32, momentum: f32) -> Self

Trait Implementations

impl Module for BatchNorm1d

fn forward(&self, input: &Tensor) -> Tensor

Forward pass

fn parameters(&self) -> Vec<Tensor>

Get all trainable parameters

fn train(&mut self)

Set module to training mode

fn eval(&mut self)

Set module to evaluation mode

fn is_training(&self) -> bool

Check if in training mode

fn num_parameters(&self) -> usize

Number of trainable parameters