1. 程式人生 > >tensorflow隨筆-佇列(11)

tensorflow隨筆-佇列(11)

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Thu Sep  6 10:16:37 2018
@author: myhaspl
@email:[email protected]
"""
 
import tensorflow as tf
import numpy as np
 
q=tf.RandomShuffleQueue(capacity=200,min_after_dequeue=2,dtypes="float")
x=np.random.randint(100,size=10)
enqueue_op=q.enqueue([x])
 

init=tf.global_variables_initializer()
qr=tf.train.QueueRunner(q,enqueue_ops=[enqueue_op]*1)
#主執行緒
with tf.Session() as sess: 
    sess.run(init)
    coord=tf.train.Coordinator()#協調器,協調執行緒間的關係。
    enqueue_threads=qr.create_threads(sess,coord=coord,start=True)#啟動入隊執行緒
    for i in range(0,10):
        print sess.run(q.dequeue())
    coord.request_stop()#通知其它執行緒關閉
    coord.join(enqueue_threads)#關閉其它執行緒後,合併到主執行緒
 
 
    
 
    
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]
[69. 16. 50. 50. 50. 76. 25. 75. 58. 77.]