代码之家  ›  专栏  ›  技术社区  ›  Ufuk Can Bicici

更改TensorFlow中tf.placeholder对象的当前图形:是否可能?

  •  0
  • Ufuk Can Bicici  · 技术社区  · 6 年前

    在我的代码中,我出于某种原因更改了当前的默认图,重新从头开始构建所有的计算逻辑。这自然会导致一些错误,因为 tf.placeholder 项目保留在旧图表中。我可以自然地再次声明这些代码,但是为了做到这一点,我必须编写大量的胶垃圾代码,这将不必要地使所有事情复杂化。我需要的是得到所有 占位符 对象在我当前的图形中,然后转移到我要创建的新图形中。有什么办法吗?我的初步研究没有给出任何有意义的结果,但我非常肯定,在TensorFlow中应该有一种方法可以做到这一点。

    1 回复  |  直到 6 年前
        1
  •  2
  •   nessuno    6 年前

    如果有两个图,可以使用 tf.contrib.copy_graph 模块。

    特别是,您可以使用 copy_op_to_graph 复制占位符 pl 从图表 g1 到图 g2 :

    tf.contrib.copy_graph.copy_op_to_graph(
        pl,
        g2,
        [],
        scope=''
    )