Python module
group_norm
Group Normalization implementation using the graph API.
GroupNorm
class max.nn.norm.group_norm.GroupNorm(num_groups, num_channels, eps=1e-05, affine=True, device=cpu:0)
Group normalization block.
Divides channels into groups and computes normalization stats per group. Follows the implementation pattern from PyTorch’s group_norm.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!