Skip to main content

BatchNorm2d

BatchNorm2d(num_features, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True, device=None, dtype=None, *, activation=None)

Batch normalization of ``[batch, height, width, channels]`` input.