torch.size(0)中的0表示第0维度的数据数量,torch.size(1)同理。
code:
a = torch.tensor([[1,2,3], [4,5,6]])
print(a.size(0)) # 2 第0维有[1,2,3]和[4,5,6]两个数据
print(a.size(1)) # 3 第1维有1,2,3(或4,5,6)三个数据
torch.size(0)中的0表示第0维度的数据数量,torch.size(1)同理。
code:
a = torch.tensor([[1,2,3], [4,5,6]])
print(a.size(0)) # 2 第0维有[1,2,3]和[4,5,6]两个数据
print(a.size(1)) # 3 第1维有1,2,3(或4,5,6)三个数据