TensorFlow学习--tf.FIFOQueue队列

来源:互联网 发布:代码优化工具 编辑:程序博客网 时间:2024/06/03 10:50

队列

tf.FIFOQueue

tensorflow使用tf.FIFOQueue类创建一个先入先出队列.
属性:
capacity:指定队列中的元素数量的上限。
dtypes:DType对象的列表。dtypes的长度必须等于每个队列元素中张量的数量。
shapes:(可选项)
names:(可选项)命名队列的字符串。
shared_name :(可选项)如果非空,则将在多个会话中以给定名称共享此队列。
name:队列操作的可选名称。

#!/usr/bin/python# coding:utf-8# 队列与多线程import tensorflow as tf# 创建一个先入先出的队列,指定队列最多可以保存3个元素,并指定类型为整数q = tf.FIFOQueue(3, 'int32')# 初始化队列中的元素,将[0,10,20]3个元素排入此队列init = q.enqueue_many(([0, 10, 20], ))# 将队列中的第1个元素出队列,并存入变量x中x = q.dequeue()# 将得到的值加1y = x + 1# 将加1后的值重新加入队列q_inc = q.enqueue([y])with tf.Session() as sess:    # 队列初始化    init.run()    for _ in range(5):        # 执行数据出队列/出队元素+1/重新加入队列的过程        v, _ = sess.run([x, q_inc])        print v

队列开始有[0,10,20]三个元素,执行5次数据出队列,出队元素+1,重新加入队列的过程中:
x=0,  y=1,   队列:[10,20,1]
x=10,  y=11,  队列:[20,1,11]
x=20,  y=21,  队列:[1,11,21]
x=1,   y=2,   队列:[11,21,2]
x=11,  y=12,  队列:[21,2,12]

阅读全文
0 0