Expand description
Batch Normalization Layer
Implements batch normalization with learned scale (gamma) and shift (beta) parameters: y = gamma * (x - mean) / sqrt(var + epsilon) + beta
During inference, uses running mean and variance computed during training.
Structs§
- Batch
Norm - Batch Normalization layer
Type Aliases§
- Batch
Norm2d - Alias for BatchNorm (for API compatibility with PyTorch naming).