pytorch 扩展张量

目录

repeat函数

expand函数：

expand

expand_as函数：

repeat函数

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      a = torch.tensor([[1, 2, 3],[1, 2, 3]])
     
    
   

    

     

    
    

     

      print(a)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      b = torch.tensor([[2, 2, 2], [3, 3, 3], [3, 3, 3], [3, 3, 3]])
     
    
   

    

     

    
    

     

      print(a.shape,b.size())
     
    
   

    

     

    
    

     

      c = a.repeat(2,1)
     
    
   

    

     

    
    

     

      print(c)
     
    
   

    

     

    
    

     

      print(c.size())
     
    
  
 

expand

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

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