pub fn batch_norm(input: &Tensor, eps: f64) -> Result<Tensor, TorshError>
Batch normalization for graphs Normalizes features across the batch dimension