Channel-wise Batch Normalization for 4D Feature Maps - Coding problem | OneNoughtOne