Struct BatchNormalization 

pub struct BatchNormalization { /* private fields */ }

Batch normalization layer

Implementations

impl BatchNormalization

pub fn new() -> Self

Create new batch normalization layer

pub fn momentum(self, momentum: f64) -> Self

Set momentum

pub fn epsilon(self, epsilon: f64) -> Self

Set epsilon

pub fn name(self, name: impl Into<String>) -> Self

Set layer name

pub fn set_training(&mut self, training: bool)

Set training mode

Trait Implementations

impl Default for BatchNormalization

fn default() -> Self

Returns the “default value” for a type.

impl KerasLayer for BatchNormalization

fn build(&mut self, input_shape: &[usize]) -> Result<()>

Build the layer (called during model compilation)

fn call(&self, inputs: &ArrayD<f64>) -> Result<ArrayD<f64>>

Forward pass through the layer

fn compute_output_shape(&self, input_shape: &[usize]) -> Vec<usize>

Compute output shape given input shape

fn name(&self) -> &str

Get layer name

fn get_weights(&self) -> Vec<ArrayD<f64>>

Get trainable parameters

fn set_weights(&mut self, weights: Vec<ArrayD<f64>>) -> Result<()>

Set trainable parameters

fn built(&self) -> bool

Check if layer is built

fn count_params(&self) -> usize

Get number of parameters