BatchNormalization
constructor(numFeatures: Int, eps: Double = 1.0E-5, momentum: Double = 0.1, affine: Boolean = true, name: String = "BatchNormalization", initGamma: Tensor<T, V>? = null, initBeta: Tensor<T, V>? = null)(source)
Parameters
numFeatures
Number of features (channels)
eps
Small value added to the denominator for numerical stability
momentum
Momentum for running statistics update during training
affine
Whether to learn affine parameters (gamma and beta)
name
Name of the module
initGamma
Initial gamma (scale) parameter
initBeta
Initial beta (shift) parameter