torch 数组

从2*1演化而来，

2行1列就是2*1

2*2*1

就是最后维度只有1个数字，这个数字

  

   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

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

    

     

    
    

     

       print(bb.shape)
     
    
   

    

     

    
    

     

       exit()
     
    
  
 

bb=bb.squeeze(2)

降个维度，就是2*2了。

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

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