pytorch: 获取 tensor 维度的方法：shape 和 size()

pycharm 获取 tensor 的方法有两种：shape 和 size()
tensor 是类 Tensor() 的实例， 其中shape是其属性，而 size() 是其继承的方法，两者均可以获得 tensor 的维度。

import torch

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

输出为：
torch.Size([2, 3])
torch.Size([2, 3])

另外，还可以获取其中的某一维度

print(a.shape[0])
print(a.shape[1])
print(a.size(0))
print(a.size(1))

输出为：

2
3
2
3

重点：
只要注意其用法就行了，shape 是属性，使用中括号，size() 是函数，使用小括号。