pub fn batch_norm(
x: &[f64],
running_mean: &[f64],
running_var: &[f64],
gamma: &[f64],
beta: &[f64],
eps: f64,
) -> Result<Vec<f64>, String>Expand description
Batch normalization (inference mode). y = gamma * (x - running_mean) / sqrt(running_var + eps) + beta.