【发布时间】:2016-11-02 00:54:09
【问题描述】:
我有一个工作输入管道,它使用多个线程从 tfrecords 文件中读取数据,对数据进行解码和预处理,并将它们提供给图形。我使用this 示例作为起点。我的问题是我需要在损失层中再次输入批次,以计算输入的重建误差。如果我现在只是将出队操作多次添加到我的图表中,即一次在输入层,一次在损失层,我显然会得到两次不同的批次,因为 tensorflow 两次评估操作。克服这个问题的最简单方法是什么?
【问题讨论】:
-
如果您在一次运行调用中需要两次 value,请将 deque 连接到
tf.identity并引用tf.identity两次。如果您在不同的运行调用中需要它,请通过运行tf.assign将出列值保存到变量并引用该变量 -
谢谢,正是我想要的,这解决了我的问题!
标签: python tensorflow