torch扩展维度

unsqueeze需要，否则报错

expand(3,2,2)参数就是目标维度。

是复制方式，最后的维度必须能整除

  

   

    

     

    
    

     

      if __name__ == '__main__':
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       x = torch.Tensor([[1,2], [2,3], [3,4]])
     
    
   

    

     

    
    

     

       print(x.size())
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       print(x)
     
    
   

    

     

    
    

     

       d=x.unsqueeze(1).expand(3,2, 2)
     
    
   

    

     

    
    

     

       print(d)
     
    
  
 

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

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