BaseNormalization

Base class for normalization layers.

Abstract Signature:

BaseNormalization(input_dim: int = 0)

PyTorch

API: torch.nn.Module
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: praxis.layers.BaseNormalization
Strategy: Direct Mapping
Official Docs ↗