【莫凡PyTorch教程筆記】-1. PyTorch 神經網絡基礎

Torch or Numpy? 用numpy還是Torch Torch 自稱爲神經網絡界的 Numpy, 因爲他能將 torch 產生的 tensor 放在 GPU 中加速運算 (前提是你有合適的 GPU), 就像 Numpy 會把 array 放在 CPU 中加速運算. 所以神經網絡的話, 當然是用 Torch 的 tensor 形式數據最好咯. 就像 Tensorflow 當中的 tensor
相關文章
相關標籤/搜索