pytorch.range() 和 pytorch.arange() 的区别

  

   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       y = torch.range(1, 6)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       print(y)
     
    
   

    

     

    
    

     

       print(y.dtype)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       z = torch.arange(0, 6)
     
    
   

    

     

    
    

     

       print(z)
     
    
   

    

     

    
    

     

       print(z.dtype)
     
    
  
 

结果：数据类型不一样，长度不一样，可以按自己的需求选择合适的

tensor([1., 2., 3., 4., 5., 6.])
torch.float32
tensor([0, 1, 2, 3, 4, 5])
torch.int64

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

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