首頁 / 名詞卡 / 資料工程與MLOps / PyTorch

iPAS AI中級🗄️ 資料工程與MLOps
🔑 名詞卡

PyTorch

🎯 定義

由 Meta(Facebook AI Research)開發的開源機器學習框架,以動態計算圖(Define-by-Run)著稱,運算在定義時立即執行,行為更像標準 Python 代碼。

重點/應用

  • 內建自動微分(Autograd)引擎,能自動計算任何運算的梯度,簡化反向傳播的複雜性。
  • Pythonic 介面,與 Python 語言習慣高度契合,簡化調試與實驗流程。
  • 生態系統:torchvision(視覺)、torchtext(NLP)、torchaudio(語音)、PyTorch Lightning、Hugging Face Transformers。

記憶技巧

「PyTorch 邊寫邊跑(動態計算圖),像即興演出」,在學術研究與快速原型開發領域迅速崛起。

易混淆

  • vs TensorFlow:PyTorch(動態計算圖,Define-by-Run)vs TensorFlow 早期版本(靜態計算圖,需先建完整圖再執行)。
📖 課文原文

「PyTorch 最大的特點是採用動態計算圖,這意味著運算在定義時立即執行,而非像早期 TensorFlow 那樣先構建完整圖形再運行。」