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

 

posted on 2018-08-08 11:38  吱吱了了  阅读(346)  评论(0编辑  收藏  举报

导航