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

最新推荐文章于 2021-02-23 00:10:37 发布

Saul Zhang

最新推荐文章于 2021-02-23 00:10:37 发布

阅读量254

点赞数

分类专栏：深度学习笔记

本文链接：https://blog.csdn.net/qq_37053885/article/details/79193921

版权

深度学习笔记专栏收录该内容

28 篇文章 2 订阅

订阅专栏

一、队列与多线程

队列是多线程数据处理框架的基础，修改队列状态的主要操作有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)