第4章 PyTorch深度學習基礎

在介紹PyTorch之前，讀者需要先了解Numpy。Numpy是一種用于科學計算的框架，它提供了一個N維矩陣對象ndarray，初始化、計算ndarray的函數，以及變換ndarray形狀和組合拆分ndarray的函數。

PyTorch的Tensor與Numpy的ndarray十分類似，但是Tensor具備兩個ndarray不具備，但是對于深度學習來說非常重要的功能：一是Tensor能用GPU計算。GPU根據芯片性能的不同，在進行矩陣運算時，能比CPU快幾十倍；二是Tensor在計算時能夠作為結點自動加入計算圖中，而計算圖可以為其中的每個結點自動計算微分。下面，我們首先介紹Tensor對象及其運算。后文給出的代碼都依賴于以下兩個模塊。