1x1卷积降维-代码

初始:[4,256,56,72]

目标:[4,128,56,72]

  1. __init__块:
    
    self.res_down1 = nn.Conv2d(in_channels=256, out_channels=128, kernel_size=1)
  2. def forward(self, x):
    res_2_2 = self.res_encoder2(res_1)              #res_1:[4,256,56,72]
    res_2 = self.res_down1(res_2_2)                 #目标:[4,128,56,72]


版权声明:本文为weixin_41904017原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。