tensor笔记3---size()

代码： 

a = torch.tensor([[1, 2, 3],
                  [4, 5, 6]])
print(a.size())
print(a.size(0))
print(a.size(1))

输出：

torch.Size([2, 3])
2
3