Tensorflow出列内while_loop

Question

考虑这个简单的例子：

l = [1,2,324,3,12,1,2,3] 
q = tf.train.input_producer(l, shuffle=False) 
x = q.dequeue() 

t = tf.TensorArray(dtype=tf.int32, size=5, dynamic_size=True, clear_after_read=True) 

_, t = tf.while_loop(cond=lambda i, a: tf.less(i, 5, name='less_op'), 
body=lambda i, a: [i+1, a.write(i, [x])], 
loop_vars=[0, t]) 

它输出[1 1 1 1 1]，由于出队（）被调用一次。请告诉我如何在每次迭代中触发出列操作。

谢谢！

干杯，克里斯

Answer 1

问题的出现是因为tf.while_loop()体捕获张x作为不变的一个循环的值 ，而你要出队要在循环内进行的副作用。

的解决方案是将呼叫转移到q.dequeue()身体内部，如下所示：

import tensorflow as tf 

l = [1, 2, 324, 3, 12, 1, 2, 3] 
q = tf.train.input_producer(l, shuffle=False) 
t = tf.TensorArray(dtype=tf.int32, size=5, dynamic_size=True, clear_after_read=True) 

# N.B. set `parallel_iterations=1` to ensure that values are dequeued in a 
# deterministic FIFO order. 
_, t = tf.while_loop(cond=lambda i, a: tf.less(i, 5, name='less_op'), 
        body=lambda i, a: [i+1, a.write(i, [q.dequeue()])], 
        loop_vars=[0, t], 
        parallel_iterations=1) 

result = t.stack() 

sess = tf.Session() 
tf.train.start_queue_runners(sess) 
print(sess.run(result)) # ==> '[[1], [2], [324], [3], [12]]' 
print(sess.run(result)) # ==> '[[1], [2], [3], [1], [2]]'