Tensor是什麼
張量,是標量、向量、矩陣的高維擴展
0維張量,標量
1維張量,向量
2維張量,矩陣
n維張量
Variable是torch.autograd中的數據類型,主要用於封裝Tensor,進行自動求導
data: 被包裝的Tensor
grad: data的梯度
grad_fn: 創建Tensor的Function, 是自動求導的關鍵
requires_grad: 指示是否需要梯度
is_leaf: 指示是否是葉子結點(張量)
Pytorch0.4.0版本開始,Variable併入Tensor
這四個屬性是數據有關的屬性
data:
dtype: 張量的數據類型,如torch.FloatTensor, torch.cuda.FloatTensor
shape: 張量的形狀,如(64, 3, 224, 224)
device: 張量所在設備,GPU/CPU是加速的關鍵
這個是求導有關的屬性
requries_grad:
grad:
grad_fn:
is_leaf:
Tensor的創建
1.直接創建
2.依據數據創建
3.依據概率創建