import tensorflow as tf
queue = tf.FIFOQueue(10, dtypes=[tf.float32])
y = tf.Variable(initial_value=1.0, dtype=tf.float32)
y_assign = tf.assign_add(y, 1)
with tf.control_dependencies([y_assign]):
enque_op = queue.enqueue([y])
dequeue_op = queue.dequeue()
ev = threading.Event()
# 声明创建5个线程,将线程管理器送到线程中
def enque_data(sess):
while True:
if not ev.is_set():
sess.run(enque_op)
print('.', end='')
else:
break
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
threads = [threading.Thread(target=enque_data, args=[sess]) for id in range(1)]
for t in threads:
t.isDaemon()
t.start()
for i in range(10):
print(sess.run(dequeue_op))
ev.set() #当主线程设置ev标志位为True时,线程执行的enque_data函数会break,执行完程序。另外,由于线程设置了t.isDaemon(),主线程执行完成,从线程会关闭。
输出:
..3.0
..5.0
.7.0
.8.0
.9.0
.10.0
..11.0
.12.0
.13.0
.14.0
.