Function layer_norm

pub fn layer_norm<A, S, D>(x: &ArrayBase<S, D>, eps: f64) -> Array<A, D>
where
    A: Float + FromPrimitive,
    D: Dimension,
    S: Data<Elem = A>,