def BatchNorm(nf, ndim=2, norm_type=NormType.Batch, **kwargs):
    "BatchNorm layer with `nf` features and `ndim` initialized depending on `norm_type`."
    return _get_norm(
        "BatchNorm", nf, ndim, zero=norm_type == NormType.BatchZero, **kwargs
    )
BatchNorm layer with `nf` features and `ndim` initialized depending on `norm_type`.
analyze/convnexts4nd/src/models/baselines/nonaka/xresnet.py