歡迎您光臨本站 註冊首頁

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

←手機掃碼閱讀     kyec555 @ 2020-06-24 , reply:0

問題,在一個程式內構建好了一個圖,執行完之後想重新使用這個圖進行計算,或者想同時在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, part2], 1)  ###原張量的第四列被刪除

 

 


[kyec555 ] 解決tensorflow 釋放圖,刪除變數問題已經有409次圍觀

http://coctec.com/docs/program/show-post-239531.html