Tensorflow：如何替换计算图中的节点？

Question

如果您有两个不相交的图，并且要将它们链接起来，请转动：

x = tf.placeholder('float')
y = f(x)

y = tf.placeholder('float')
z = f(y)

进入这个：

x = tf.placeholder('float')
y = f(x)
z = g(y)

有办法吗？在某些情况下，它似乎可以使施工变得更容易。

例如，如果您有一个图形，它的输入图像是一个tf.placeholder，并且想要优化输入图像，deep dream样式，那么有没有办法用一个tf.variable节点替换占位符呢？或者在建立图表之前，你必须考虑这个问题吗？

Answer 1

TL；DR：如果您可以将这两个计算定义为Python函数，那么您应该这样做。如果不能，则在TensorFlow中有更高级的功能来序列化和导入图，这允许您从不同的源合成图。

在TensorFlow中，一种方法是将不相交的计算构建为单独的tf.Graph对象，然后使用^{}将它们转换为序列化的协议缓冲区：

with tf.Graph().as_default() as g_1:
  input = tf.placeholder(tf.float32, name="input")
  y = f(input)
  # NOTE: using identity to get a known name for the output tensor.
  output = tf.identity(y, name="output")

gdef_1 = g_1.as_graph_def()

with tf.Graph().as_default() as g_2:  # NOTE: g_2 not g_1       
  input = tf.placeholder(tf.float32, name="input")
  z = g(input)
  output = tf.identity(y, name="output")

gdef_2 = g_2.as_graph_def()

然后您可以使用^{}将gdef_1和gdef_2组合成第三个图：

with tf.Graph().as_default() as g_combined:
  x = tf.placeholder(tf.float32, name="")

  # Import gdef_1, which performs f(x).
  # "input:0" and "output:0" are the names of tensors in gdef_1.
  y, = tf.import_graph_def(gdef_1, input_map={"input:0": x},
                           return_elements=["output:0"])

  # Import gdef_2, which performs g(y)
  z, = tf.import_graph_def(gdef_2, input_map={"input:0": y},
                           return_elements=["output:0"]