Implement Batch Normalization for BCHW InputDeep-ML | Problem