Tensorflow：空计算图和垃圾收集

Question

1 个空图错误

嘿，我正在尝试完全分开运行多个张量流图，但遇到了以下继承问题。

还有

import tensorflow as tf


class A:

    g = tf.Graph()
    g.as_default()
    s = tf.Session(graph=g)

    x = tf.placeholder(tf.float32)

    __call__ = lambda self,X: self.s.run(self.y, {self.x:X})


class B(A):

    y = 2 * A.x


test = B()
print test([1,1,2])

错误

RuntimeError: The Session graph is empty.  Add operations to the graph before calling run()

---

2 - 垃圾收集

我也很想知道删除这些不同的图，如果我用 Session().close() 关闭会话并且它是唯一知道该图的会话，该图现在会消失并被垃圾收集吗？

Answer 1

问题 1

如果您希望将您的操作添加到特定图表中，您需要使用with g.as_default() 作为上下文：

class A:

    g = tf.Graph()
    with g.as_default():
        x = tf.placeholder(tf.float32)

    s = tf.Session(graph=g)

    __call__ = lambda self,X: self.s.run(self.y, {self.x:X})


class B(A):

    with g.as_default():
        y = 2 * A.x


test = B()
print test([1,1,2])

（PS：你的代码写得很糟糕，希望只是为了测试）

---

问题 2

图表不受相应会话的影响。

您可以创建图表，打开会话并关闭它，图表将保持不变：

g = tf.Graph()
with g.as_default():
    # build graph...
    x = tf.constant(1)

sess = tf.Session(graph=g)
sess.run(x)
sess.close()

# Now we can create a new session with the same graph
sess2 = tf.Session(graph=g)
sess2.run(x)
sess2.close()