Skip to main content

Module batchnorm

Module batchnorm 

Source
Expand description

Batch normalization.

Structsยง

BatchNorm1d
Batch normalisation over a 2-D [N, C] or 3-D [N, C, L] input.
BatchNorm2d
Batch normalisation over a 4-D [N, C, H, W] input.
BatchNorm3d
Batch normalisation over a 5-D [N, C, D, H, W] input.