torch.nn.GroupNorm:将channel切分成许多组进行归一化
torch.nn.GroupNorm(num_groups,num_channels)
num_groups:组数
num_channels:通道数量
代码示例:
a=torch.randn(15,256,9,15)
#将channel256分为8组,每组32channel
m=nn.GroupNorm(8,256) # 这其实一个函数
print(m(a).shape) # [15, 256, 9, 15]
输出的size和输入相同
原理:https://blog.csdn.net/qq_23981335/article/details/106572171
————————————————
版权声明:本文为CSDN博主「江南汪」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_47156261/article/details/116896439
torch.nn.GroupNorm(num_groups,num_channels)
输入通道被分成num_groups组,每个组包含num_channels/num_groups个通道,每组的均值的均值和标准差分开计算。