带有num_epochs的string_input_producer抛出OutofRangeError

Question

此代码按预期将元素出队：

import tensorflow as tf

with tf.Session() as sess:
        queue = tf.train.string_input_producer([str(i) for i in range(10)])
        deq = queue.dequeue()

        coord = tf.train.Coordinator()
        threads = tf.train.start_queue_runners(coord=coord)
        for i in range(3):
            print(sess.run([deq]))

但是，当我将num_epochs添加到string_input_producer时，出现OutofRangeError失败

import tensorflow as tf

with tf.Session() as sess:
        queue = tf.train.string_input_producer([str(i) for i in range(10)], num_epochs=1)
        deq = queue.dequeue()

        coord = tf.train.Coordinator()
        threads = tf.train.start_queue_runners(coord=coord)
        for i in range(3):
            print(sess.run([deq]))

我使用的是tensorflow 0.9

Answer 1

它需要tf.initialize_all_variables().run()