TensorFlow+实战Google深度学习框架学习笔记（7）-----队列与多线程

一、创建一个队列：

FIFOQueue：先进先出

RandomShuffleQueue：会将队列中的元素打乱，每次出列操作得到的是从当前队列所有元素中随机选择的一个。

二、操作一个队列的函数：

enqueue、enqueue_many、dequeue

import tensorflow as tf

#创建一个先进先出队列，指定队列中最多可以保存两个元素，并指定类型为整数

q = tf.FIFOQueue （ 2, "int32" )

#使用enqueue_many 函数来初始化队列中的元素。

 init = q.enueue_many( ( [0,10],))

#使用Dequeue函数将队列中的第一个元素出队列

x = q.dequeue()

#入列

q_inc = q.enqueue([x])

#将x出列又入列

with tf.Session() as sess:

     sess.run( [x,q_inc])

三、多线程协同：

tf.Coordinator :

用于协同多个线程一起停止，并提供了should_stop、request_stop、join三个函数。在启动线程之前，需要先声明一个tf.Coordinator类中提供的should_stop函数，当这个函数的返回值为True时，则当前线程也需要退出。每一个启动的线程都可以通过调用request_stop函数来通知其他线程退出。当某一个线程调用request_stop函数之后，should_stop函数的返回值将被设置为True，这样其他的线程就可以同时终止了。

tf.QueueRunner

巴特西

TensorFlow+实战Google深度学习框架学习笔记（7）-----队列与多线程

一、创建一个队列：

二、操作一个队列的函数：

三、多线程协同：

最新文章

热门文章