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