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.依据概率创建