【文檔學習】PyTorch——torch包

本系列記錄了博主學習PyTorch過程中的筆記。本文記錄的是torch包的相關內容,官方網址。只記錄了一部分博主用到的,持續更新。更新於2019.03.21。 torch包包括了用於多維張量的數據結構,並定義了對於它們的數學計算。此外,這個包也提供了多種有效的張量及任意類型的並行化用法,以及其他有用的方法。 這個包對應於CUDA,允許用戶在NVIDIA GPU(運算能力>=3.0)上運行張量計算。
相關文章
相關標籤/搜索