如何從張量流圖中刪除變量？或者如何刪除圖表並釋放內存？

Question

在我的算法中，我需要迭代地創建一個圖並使用共享變量。 問題是，我每次創建圖形並使用共享變量時，Tensorflow都會創建新變量併爲其分配內存。這會導致已用內存的增長。顯然在某些時候它會消耗所有的內存並被殺死。

因此，解決辦法要麼是這些：

1-刪除前一個迭代中創建的變量和釋放內存。然後創建新圖。但是如何？

2-每次迭代後刪除圖形和創造，釋放內存，並創建一個新的。但是接下來我需要一些如何將變量存儲在某處，以便我可以使用它們來創建下一個圖（因爲變量是共享的）。但是如何？

我該怎麼辦？

Answer 1

TensorFlow支持復位會話。有關如何使用它，請參閱reset in the session management。