Batch Group Layer Normalization Pytorch