groupNorm

open override fun groupNorm(numGroups: Int, numChannels: Int, eps: Double, affine: Boolean, id: String)(source)

Creates a group normalization layer - alternative normalization approach. Normalizes the input by dividing channels into groups and normalizing within each group.

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)

id

Optional identifier for the layer