BatchNorm layer with `nf` features and `ndim` initialized depending on `norm_type`.
BatchNorm(
nf,
ndim = 2,
norm_type = 1,
eps = 1e-05,
momentum = 0.1,
affine = TRUE,
track_running_stats = TRUE
)input shape
dimension number
normalization type
epsilon
momentum
affine
track running statistics
None