解决tensorflow 释放图,删除变量问题

砸漏

发布于 2020-10-21 15:10:29

9530

发布于 2020-10-21 15:10:29

文章被收录于专栏：恩蓝脚本

问题，在一个程序内构建好了一个图，运行完之后想重新使用这个图进行计算，或者想同时在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)
###原张量的第四列被删除

以上这篇解决tensorflow 释放图,删除变量问题就是小编分享给大家的全部内容了，希望能给大家一个参考。

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2020-09-11 ，如有侵权请联系 cloudcommunity@tencent.com 删除

tensorflow