python-tensorflow 多线程+Event进入enqueue

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
.

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值