Tensorflow--多线程数据处理框架

一、队列与多线程

队列是多线程数据处理框架的基础,修改队列状态的主要操作有Enqueue、EnqueueMany和Dequeue

以下程序展示了FIFOQueue队列的基本操作:

import tensorflow as tf 
#创建一个先进先出的队列,指定队列中最多保存两个元素,并指定类型为整型
q = tf.FIFOQueue(2,"int32")

init = q.enqueue_many(([0,10],))
x = q.dequeue()
y = x + 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)


猜你喜欢

转载自blog.csdn.net/qq_37053885/article/details/79193921