Skip to main content

Module batchnorm

Module batchnorm 

Source
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§

BatchNorm
Batch Normalization layer

Type Aliases§

BatchNorm2d
Alias for BatchNorm (for API compatibility with PyTorch naming).