1. 程式人生 > 程式設計 >解決tensorflow 釋放圖,刪除變數問題

解決tensorflow 釋放圖,刪除變數問題

問題,在一個程式內構建好了一個圖,執行完之後想重新使用這個圖進行計算,或者想同時在train完的時候做test,就會提示***變數已存在。

解決辦法:

待一個圖執行完之後加上tf.reset_default_graph來重置預設的圖。

補充知識:tensorflow 張量插入或者刪除的辦法

我就廢話不多說了,大家還是直接看程式碼吧~

part1 = new_enc_in[:,0:3 ]
part2 = new_enc_in[:,3:]
###part_new 要插入的張量
new_enc_in = tf.concat([part1,part_new,part2],1)
##得到的張量裡part_new為新插入列
 
##如果要刪除,則修改part2的範圍
part1 = new_enc_in[:,0:3]
part2 = new_enc_in[:,4:]
new_enc_in = tf.concat([part1,1)
###原張量的第四列被刪除

以上這篇解決tensorflow 釋放圖,刪除變數問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。