【PyTorch学习笔记】1.Tensor 与 Variable

在 PyTorch 0.4.0 之前，torch.autograd 包中存在 Variable 这种数据类型，主要是用于封装 Tensor，进行自动求导。Variable 主要包含下面几种属性。

 

• data: 被包装的 Tensor。
• grad: data 的梯度。
• grad_fn: 创建 Tensor 所使用的 Function，是自动求导的关键，因为根据所记录的函数才能计算出导数。
• requires_grad: 指示是否需要梯度，并不是所有的张量都需要计算梯度。

在 PyTorch 0.4.0 之后，Variable 并入了 Tensor。在之后版本的 Tensor 中，除了具有上面 Variable 的 5 个属性，还有另外 3 个属性。

 

• dtype: 张量的数据类型，如 torch.FloatTensor，torch.cuda.FloatTensor。
• shape: 张量的形状。如 (64, 3, 224, 224)
• device: 张量所在设备 (CPU/GPU)，GPU 是加速计算的关键

 

 参考自1.2 Tensor(张量)介绍 - PyTorch 学习笔记 (zhangxiann.com)