pub fn trunc_normal_init<F: Float>(std: F, a: F, b: F) -> impl Fn() -> F
Truncated normal initialization