torch数据维度

3,1 表示最外层括号中有3条数据，2个逗号

torch的数据维度表示，从外向里表示，[3,1]，表示外面有3行1列

  

   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     
 import torch.nn as nn
     
    
   

    

     

    
    

     
 import torch.nn.functional as F
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       data = torch.randn(3,1,6)
     
    
   

    

     

    
    

     

       data2 = torch.randn(3, 1)
     
    
   

    

     

    
    

     

       print(data2.size())
     
    
   

    

     

    
    

     

       print(data2)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       data2 = torch.randn(1, 3)
     
    
   

    

     

    
    

     

       print(data2.size())
     
    
   

    

     

    
    

     

       print(data2)
     
    
  
 

torch.Size([3, 1])
tensor([[ 1.5015],
        [-1.8209],
        [ 0.9194]])
torch.Size([1, 3])
tensor([[ 1.4510, -1.3262, -0.5420]])

target的第2维和data的第3维数量要保持一致：否则报错：

  

   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     
 import torch.nn as nn
     
    
   

    

     

    
    

     
 import torch.nn.functional as F
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       data = torch.randn(3,3,2)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       target = torch.tensor([[1,1],[0,1],[0,1]])
     
    
   

    

     

    
    

     

       print(data.size(),target.size())
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       n = data.size(0)
     
    
   

    

     

    
    

     

       c = data.size(1)
     
    
   

    

     

    
    

     

       out_size =
     
    
  
 

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

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