pub fn xavier_normal(shape: &[usize], fan_in: usize, fan_out: usize) -> Tensor
Initialize tensor with Xavier/Glorot normal