nn.ReflectionPad2d()

ReflectionPad2d()   类似于一种 镜像填充  

说说公式，及属性   具体官方文档很详细  不过不够通俗= =

 

官方文档上 是这个     

表示第一眼  有点不理解= =    

但是输出是   

 
 

   
   

    
    

   
   
   
   

    
    

     
     H(
     
     out) = H(
     
     in) + paddingTop + 
     
     paddingBottom
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
     W(
     
     out) = W(
     
     in) + paddingLeft + paddingRight 
    
    
   
   
 
 

直接来例子试试    最重要的是下面 # laft, right, top, bottom  这个注释

 
 

   
   

    
    

   
   
   
   

    
    

     
     input = torch.randn(
     
     64, 
     
     3, 
     
     220, 
     
     220) 
     
     # input size
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
      
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
     # 输入4-tuple
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
     pad = nn.ReflectionPad2d((
     
     3, 
     
     3, 
     
     5, 
     
     5))     
     
     # laft, right, top, bottom
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
     output = pad(input) 
     
     # size(64, 3, 230, 226)
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
      
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
     # 输入int
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
     pad = nn.ReflectionPad2d(
     
     3)
    
    
   
   

   
   

    
    

   
   
   
   

    
    

     
     output = pad(input) 
     
     # size(64, 3, 226, 226)
    
    
   
   
 
 

通过运算 发现的确是这样的结果      所以记住公式就行了