GroupNormalization
constructor(numGroups: Int, numChannels: Int, eps: Double = 1.0E-5, affine: Boolean = true, name: String = "GroupNormalization", initGamma: Tensor<T, V>? = null, initBeta: Tensor<T, V>? = null)(source)
Parameters
numGroups
Number of groups to divide the channels into
numChannels
Number of channels in the input
eps
Small value added to the denominator for numerical stability
affine
Whether to learn affine parameters (gamma and beta)
name
Name of the module
initGamma
Initial gamma (scale) parameter
initBeta
Initial beta (shift) parameter