Struct BatchNorm 

pub struct BatchNorm { /* private fields */ }

Batch normalization layer.

Implementations

impl BatchNorm

pub fn new( name: &str, scale: Box<dyn ArrayProtocol>, offset: Box<dyn ArrayProtocol>, running_mean: Box<dyn ArrayProtocol>, running_var: Box<dyn ArrayProtocol>, epsilon: f64, ) -> Self

Create a new batch normalization layer.

pub fn withshape( name: &str, num_features: usize, epsilon: Option<f64>, _momentum: Option<f64>, ) -> Self

Create a new batch normalization layer with initialized parameters.

Trait Implementations

impl Layer for BatchNorm

fn layer_type(&self) -> &str

Forward pass through the layer. Get the layer type name for serialization.

fn forward( &self, inputs: &dyn ArrayProtocol, ) -> Result<Box<dyn ArrayProtocol>, OperationError>

fn parameters(&self) -> Vec<Box<dyn ArrayProtocol>>

Get the layer’s parameters.

fn parameters_mut(&mut self) -> Vec<&mut Box<dyn ArrayProtocol>>

Get mutable references to the layer’s parameters.

fn update_parameter( &mut self, name: &str, value: Box<dyn ArrayProtocol>, ) -> Result<(), OperationError>

Update a specific parameter by name

fn parameter_names(&self) -> Vec<String>

Get parameter names

fn train(&mut self)

Set the layer to training mode.

fn eval(&mut self)

Set the layer to evaluation mode.

fn is_training(&self) -> bool

Check if the layer is in training mode.

fn name(&self) -> &str

Get the layer’s name.