PyTorch入門(四)——計算圖與動態圖機制

計算圖與動態圖機制

計算圖

定義:計算圖是用來描述運算的有向無環圖
計算圖中的主要元素:

  1. 結點:表示數據(向量,矩陣,張量)
  2. 邊:表示運算(加減乘除卷積)

有了計算圖之後,就可以根據計算圖進行求導

葉子結點:用戶創建的結點稱爲葉子結點
其中Tensor中屬性is_leaf指示了張量是否爲葉子結點
grad_fn記錄了創建張量時所用的方法(函數,操作)

動態圖

根據計算圖的搭建方式可以分爲:動態圖靜態圖

  1. 動態圖:運算與搭建同時進行
    特點:靈活,易調節——PyTorch
  2. 靜態圖:先搭圖,後運算
    特點:高效,不靈活——TensorFlow
發佈了49 篇原創文章 · 獲贊 45 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章