卷积分组

groups是输入和输出的公约数

  

   

    

     

    
    

     

       input = torch.Tensor(1, 48, 112, 112)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       conv_23=Conv_block(48, 32, groups=16, kernel=3, padding=1, stride=1)
     
    
   

    

     

    
    

     
 # conv_23 = Depth_Wise(64, 32, kernel=(3, 3), stride=(2, 2), padding=(1, 1), groups=128)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       x = conv_23(input)
     
    
   

    

     

    
    

     

       print(x.size())
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/114271484