当前位置: 首页 > 后端技术 > Python

TensorFlow队列多线程

时间:2023-03-25 19:54:28 Python

1.创建队列Q=tf.FIFOQueue(10000,tf.float32)10000:queuesizetf.float32:queuedatatype2.创建数据var=tf.Variable(0.0)data=tf.assign_add(var,tf.constant(1.0))3.将数据放入队列en_q=Q.enqueue(data)4.创建队列管理器qr=tf.train.QueueRunner(Q,enqueue_ops=[en_q]*2)5.创建线程管理器coor=tf.train.Coordinator()6.开启线程threads=qr.create_threads(sess,coord=coord,start=True)7.回收线程coord.request_stop()coord.join()完成代码将tensorflow导入为tfQ=tf.FIFOQueue(1000,tf.float32)var=tf.Variable(0.0)data=tf.assign_add(var,tf.constant(1.0))en_q=Q.enqueue(data)qr=tf。train.QueueRunner(Q,enqueue_ops=[en_q]*2)init=tf.initialize_all_variables()withtf.Session()assess:sess.run(init)coord=tf.train.Coordinator()threads=qr.create_threads(sess,coord=coord,start=True)foriinrange(200):打印(sess.run(Q.dequeue()))coord.request_stop()coord.join()