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