Solved by ptrblck in post #2 
              
              
                No, in this case you would use L channels. 
However, the other approache (nn.BatchNorm2d vs. nn.BatchNorm1d) should yield the same result. Since you are using the affine batchnorm transformation, you would have to make sure the weight parameter is set to equal values (bias should be all zeros in bo…