GroupNorm

Normalize groups along the channel dimension, independent of the batch size.

torch.nn.GroupNorm(
    num_groups: int,
    num_channels: int,
    eps: float = 1e-5,
    affine: bool = True,
    device=None,
    dtype=None
)

**Table 1** Parameter description
Parameter	Type	Mandatory (Yes/No)	Description
num_groups	int	Yes	Number of groups (must be divisible by num_channels)
num_channels	int	Yes	Total number of input channels
eps	float	No	Numerical stability constant (default value: 1e-5)
affine	bool	No	Whether to learn the scaling factor γ and shift β (default value: True)

Parent topic: Operator Interfaces