BatchNorm2d — PyTorch 1.11.0 documentation
pytorch.org › docs › stableBatchNorm2d. Applies Batch Normalization over a 4D input (a mini-batch of 2D inputs with additional channel dimension) as described in the paper Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift . \beta β are learnable parameter vectors of size C (where C is the input size). By default, the elements of.