torch.nn.GroupNorm用法
torch.nn.GroupNorm：将channel切分成许多组进行归一化torch.nn.GroupNorm(num_groups,num_channels)num_groups:组数num_channels:通道数量代码示例：a=torch.randn(15,256,9,15)#将channel256分为8组，每组32channelm=nn.GroupNorm(8,256)...
·
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)
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
9
0- 0
已为社区贡献11条内容
所有评论(0)