Riemannian approach to batch normalization