GroupNorm#
- class GroupNorm(num_groups, num_channels, eps=1e-5, affine=True, **kwargs)[源代码]#
Simple implementation of GroupNorm. Only support 4d tensor now. Reference: https://arxiv.org/pdf/1803.08494.pdf.
Simple implementation of GroupNorm. Only support 4d tensor now. Reference: https://arxiv.org/pdf/1803.08494.pdf.