BatchNorm2d — PyTorch 1.10.1 documentation
pytorch.org › docs › stableBecause the Batch Normalization is done over the C dimension, computing statistics on (N, H, W) slices, it’s common terminology to call this Spatial Batch Normalization. Parameters. num_features – C C C from an expected input of size (N, C, H, W) (N, C, H, W) (N, C, H, W) eps – a value added to the denominator for numerical stability. Default: 1e-5