airs_layers/normalization_layer/
mod.rs

1use airs_types::{Layer, Tensor};
2
3pub struct BatchNorm {
4    pub gamma: Tensor,
5    pub beta: Tensor,
6    pub mean: Tensor,
7    pub variance: Tensor,
8    pub epsilon: f64,
9    pub momentum: f64,
10    pub is_training: bool,
11}
12
13impl Layer for BatchNorm {
14    fn forward(&self, xs: &Tensor) -> Tensor {
15        todo!()
16    }
17}

airs_layers/normalization_layer/mod.rs

airs_layers/normalization_layer/
mod.rs