脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Python - 解决tensorflow 释放图,删除变量问题

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

2020-06-24 09:42HY_JT Python

这篇文章主要介绍了解决tensorflow 释放图,删除变量问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

问题,在一个程序内构建好了一个图,运行完之后想重新使用这个图进行计算,或者想同时在train完的时候做test,就会提示***变量已存在。

解决办法:

待一个图运行完之后加上tf.reset_default_graph来重置默认的图。

补充知识:tensorflow 张量插入或者删除的办法

我就废话不多说了,大家还是直接看代码吧~

?
1
2
3
4
5
6
7
8
9
10
11
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 释放图,删除变量问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/HY_JT/article/details/81363634

延伸 · 阅读

精彩推荐