Function batch_norm 

pub fn batch_norm(
    tensor: &DenseTensor,
    mean: &DenseTensor,
    var: &DenseTensor,
    epsilon: f64,
) -> DenseTensor

Batch Normalization（简化版）