第4章 PyTorch深度学习基础

在介绍PyTorch之前，读者需要先了解Numpy。Numpy是一种用于科学计算的框架，它提供了一个N维矩阵对象ndarray，初始化、计算ndarray的函数，以及变换ndarray形状和组合拆分ndarray的函数。

PyTorch的Tensor与Numpy的ndarray十分类似，但是Tensor具备两个ndarray不具备，但是对于深度学习来说非常重要的功能：一是Tensor能用GPU计算。GPU根据芯片性能的不同，在进行矩阵运算时，能比CPU快几十倍；二是Tensor在计算时能够作为结点自动加入计算图中，而计算图可以为其中的每个结点自动计算微分。下面，我们首先介绍Tensor对象及其运算。后文给出的代码都依赖于以下两个模块。