解決tensorflow 釋放圖,刪除變數問題
阿新 • • 發佈:2020-06-24
問題,在一個程式內構建好了一個圖,執行完之後想重新使用這個圖進行計算,或者想同時在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 釋放圖,刪除變數問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。