pub fn normal_init( shape: (usize, usize), mean: f64, std: f64, rng: &mut Random, ) -> Array2<f64>
Initialize embeddings with normal distribution