首页 文章

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

提问于
浏览
0

在我的代码中,由于某种原因,我正在更改当前的默认图形,并从头开始重建所有计算逻辑 . 这自然会导致一些错误,因为我的 tf.placeholder 项目保留在旧图表中 . 我可以自然地再次声明这些,但为了做到这一点,我必须编写很多胶水垃圾代码,这将使所有内容复杂化 . 我需要的是在我当前的图形中获取所有 tf.placeholder 对象,然后将它们转移到我要创建的新图形中 . 有没有办法做到这一点?我的初步研究没有给出任何有意义的结果,但我非常积极地应该有一种方法在Tensorflow中做到这一点 .

1 回答

  • 2

    如果有2个图形,则可以使用tf.contrib.copy_graph模块将操作和变量从图形复制到另一个图形 .

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

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

相关问题