教材:《TensorFlow实战:Google深度学习框架》
章节:7.3.1 队列与多线程
学习使用TensorFlow中的队列,代码非常简单:
import tensorflow as tf
# 创建一个先进先出队列,指定队列中最多可以保存两个元素,并指定类型为整数
q = tf.FIFOQueue(2, 'int32')
# 使用enqueque_many函数来初始化队列中的元素。name=None
# 和初始化变量类似,在使用队列之前需要明确地调用这个初始化过程
init = q.enqueue_many(([0, 10],))
#使用dequeque函数将队列中的第一个元素出队列,存在变量x中
x = q.dequeue()
y = x + 1
#将+1后的值重新加入队列
q_inc = q.enqueue([y])
with tf.Session() as sess:
init.run()
for _ in range(5):
v, _ = sess.run([x, q_inc])
print(v)