attribute lookup is not defined on builtin

attribute lookup is not defined on builtin

  

   

    

     

    
    

     

       @torch.jit.script_method
     
    
   

    

     

    
    

     
 def channel_shuffle(self, x, groups):
     
    
   

    

     

    
    

     

       batchsize, num_channels, height, width = x.data.size()
     
    
   

    

     

    
    

     
 assert (num_channels % groups == 0)
     
    
   

    

     

    
    

     

       channels_per_group = num_channels // groups
     
    
   

    

     

    
    

     
 # reshape
     
    
   

    

     

    
    

     

       x = x.view(batchsize, groups, channels_per_group, height, width)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 # transpose
     
    
   

    

     

    
    

     
 # - contiguous() required if transpose() is used before view().
     
    
   

    

     

    
    

     
 # See https://github.com/pytorch/pytorch/issues/764
     
    
   

    

     

    
    

     

       x = torch.transpose(x, 1, 2).contiguous()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 # flatten
     
    
   

    

     

    
    

     

       x = x.view(batchsize, -1, height, width)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 return x
     
    
  
 

修改方法：

 batchsize, num_channels, height, width = x.size()

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

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