動手學深度學習PyTorch版---筆記彙總

pytorch函數 PyTorch的Tensor,它可以是零維(又稱爲標量或一個數)、一維、二維及多維的數組。Tensor自稱爲神經網絡界的Numpy,它與Numpy相似,二者可以共享內存,且之間的轉換非常方便和高效。不過它們也有不同之處,最大的區別就是Numpy會把ndarray放在CPU中進行加速運算,而由Torch產生的Tensor會放在GPU中進行加速運算(假設當前環境有GPU)。 對Te
相關文章
相關標籤/搜索