2016-05-27 85 views
6

我試圖用深度學習模型來訓練一個實驗。 我發現tensorflow是做到這一點的最好方法。 但是有一個問題,tensorflow需要在python中寫入。 而我的計劃包含很多loops.Like這個..如何使用張量流在C++中訓練模型?

for i=1~2000 
for j=1~2000 

我知道這是蟒蛇一個很大的缺陷。 它比c慢。 我知道tensorfow有一個C++ API,但目前還不清楚。 https://www.tensorflow.org/api_docs/cc/index.html (這是我見過的最差規格) 有人能給我一個簡單的例子嗎? 我需要的只是兩個簡單的代碼。 一個是如何創建一個圖。 另一種是如何加載此圖並運行它。 我真的很渴望這個。希望有人能幫助我。

+0

幫你跑在TensorFlow的計算是不一樣的蟒蛇做,即使你從蟒蛇叫TensorFlow。你建立圖表,然後執行它。但執行發生在TensorFlow本身,速度非常快。它不使用解釋的python,除了'膠水'代碼。 – user20160

+1

我知道你的意思了。深層模型代碼足夠快。但是我有另一個代碼,它必須在許多循環中寫入(增強學習)。它是一個遞歸函數。 – darren1231

+0

我明白了。希望C++能爲你工作。如果不得不使用python API,可能會加速使用cython,作爲最後的手段。 – user20160

回答

4

這並不容易,但它是可能的。 首先,您需要在python中創建tensorflow圖並將其保存在文件中。 這篇文章可以幫助你
https://medium.com/jim-fleming/loading-a-tensorflow-graph-with-the-c-api-4caaff88463f#.krslipabt

其次,你需要編譯libtensorflow,將其鏈接到你的程序(你需要tensorflow頭爲好,所以這是一個有點棘手),並從文件加載圖形。 這篇文章可能這個時候
https://medium.com/jim-fleming/loading-tensorflow-graphs-via-host-languages-be10fd81876f#.p9s69rn7u

相關問題