pub fn normal(shape: &[usize], mean: f32, std: f32) -> Tensor
Initialize tensor with normal distribution