PyTorch | torch.zeros()怎么使用？ torch.view(-1)如何使用？

我不喜欢讲解，就喜欢直观，所以就看例子就行！ 

  

   

    

     

    
    

     

      torch.zeros(2,3,4)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      #输出：
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      tensor([[[0., 0., 0., 0.],
     
    
   

    

     

    
    

     

       [0., 0., 0., 0.],
     
    
   

    

     

    
    

     

       [0., 0., 0., 0.]],
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       [[0., 0., 0., 0.],
     
    
   

    

     

    
    

     

       [0., 0., 0., 0.],
     
    
   

    

     

    
    

     

       [0., 0., 0., 0.]]])
     
    
  
 

view是自动计算匹配的！

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      a = torch.arange(0,20)		#此时a的shape是(1,20)
     
    
   

    

     

    
    

     

      a.view(4,5).shape		#输出为(4,5)
     
    
   

    

     

    
    

     

      a.view(-1,5).shape		#输出为(4,5)
     
    
   

    

     

    
    

     

      a.view(4,-1).shape		#输出为(4,5)
     
    
   

    

     

    
    

     

      a.view(10,-1).shape		#输出为(10,2)
     
    
  
 

文章来源: kings.blog.csdn.net，作者：人工智能博士，版权归原作者所有，如需转载，请联系作者。

原文链接：kings.blog.csdn.net/article/details/109479111