1. 程式人生 > 其它 >(轉)pytorch和torch框架對比(區別 聯絡)

(轉)pytorch和torch框架對比(區別 聯絡)

一 區別
1.1 程式語言
pytorch 採用 python 語言。程式設計實際是 C語言做底層和C++做介面
torch 採用 lua 語言。程式設計實際是是 C語言做底層和lua語言做介面的深度學習庫
知識擴充套件: Lua相當於一個小型加強版的C語言,支援類和麵向物件

1.2 依賴庫
pytorch 可呼叫python強大的第三方庫,比如 opencv
torch 可呼叫 lua 庫函式,目前 lua庫函式沒有python多
pytorch 依賴庫多於 torch

1.3 效率
python的debug功能比lua強大很多,所以 pytorch 效率大於 torch

1.4 模型和中間變數的關係
pytorch中中間變數都存在於計算圖中,所有model共享中間變數
torch中的中間變數則在每一個模組中,想要呼叫其他模組的引數就必須首先複製這個模組然後再呼叫

1.5 編寫模型
pytroch:可直接修改 forward 函式
torch:不建議直接修改forward 函式,通常是修改updateOutput函式來實現forward操作

二 聯絡
2.1 程式設計
torch 程式碼在Pytorch中照樣可以編譯使用

2.2 編寫模型
用autograd技術實現自動求導,不需要再去麻煩地寫反向傳播的計算函式

2.2 總結
pytorch可以說是torch的python版,並增加了很多新功能

三 官網圖


四 兩種深度學習框架選擇
新手小白,入門推薦 pytorch
如果想學紮實,兩種框架pytorch torch 同時使用,難度係數也不高

轉自:https://blog.csdn.net/WJ_MeiMei/article/details/88720146