如何防止FIFO队列暂停有条件

Question

的FIFO Queue将在其enqueue和dequeue功能暂停时分别排队是满或空。当你只有一个队列时，这不是问题。

我的情况是我有两个队列，我有以下一些条件入队：

run_options = tf.RunOptions(timeout_in_ms=10000) 
i1, l1, i2, l2 = produce_sample() 

if l1 == l2: 
    sess.run(enqueue_same_op, feed_dict={x1: i1, y1: l1, x2: i2, y2: l2}, \ 
      options=run_options) 
else: 
    sess.run(enqueue_diff_op, feed_dict={x1: i1, y1: l1, x2: i2, y2: l2}) 

enqueue_same_op和enqueue_diff_op是分别queue_same和queue_diff操作。

因为l1<>l2很多的概率较大比l1==l2。因此，即使我采用了run_options，会议也会暂停排队queue_diff。

有没有办法让获得队列的大小或测试如果它已满？

非常感谢您提前。

Answer 1

有没有办法获得队列的大小，或测试它是否已满？

您可以使用size()功能和测试针对队列容量

capacity=1000 # your queue capacity 
dtype=tf.float32 
queue_diff = tf.FIFOQueue(capacity, dtype) 
enqueue_op = tf.cond(tf.equal(queue_diff.size(), capacity), lambda:is_full(), lambda: enqueue(queue_diff, enqueue_elements..)) 

# a function to takecare when the queue is full 
def is_full(): 
    return 'Enqueue_Failed' 

# a function for enqueue ops 
def enqueue(queue, element...): 
    queue.enqueue(element) 
    return 'Enqueue_Success' 

Answer 2

您可以通过调用size方法，它返回一个张量得到一个队列的大小设置队列的大小：

$ queue = tf.FIFOQueue(capacity=100, dtype=tf.float32) 
$ queue.size() 
<tf.Tensor 'fifo_queue_Size:0' shape=() dtype=int32> 

如果有一个固定的大小的队列，在像上述的示例，可以使用调节功能tf.cond的流量，从而判断是否满了。

或者，您可以使用capacity=-1使您的队列不受限制。严格地说，这不是开放API和官方文件中没有说明，但可以在源代码中找到它：

def _fifo_queue_v2(component_types, shapes=None, capacity=None, 
        container=None, shared_name=None, name=None): 
    r"""A queue that produces elements in first-in first-out order. 

    Args: 
    component_types: A list of `tf.DTypes` that has length `>= 1`. 
     The type of each component in a value. 
    shapes: An optional list of shapes (each a `tf.TensorShape` or list of `ints`). Defaults to `[]`. 
     The shape of each component in a value. The length of this attr must 
     be either 0 or the same as the length of component_types. If the length of 
     this attr is 0, the shapes of queue elements are not constrained, and 
     only one element may be dequeued at a time. 
    capacity: An optional `int`. Defaults to `-1`. 
     The upper bound on the number of elements in this queue. 
     Negative numbers mean no limit. 
    container: An optional `string`. Defaults to `""`. 
     If non-empty, this queue is placed in the given container. 
     Otherwise, a default container is used. 
    shared_name: An optional `string`. Defaults to `""`. 
     If non-empty, this queue will be shared under the given name 
     across multiple sessions. 
    name: A name for the operation (optional). 

    Returns: 
    A `Tensor` of type `resource`. The handle to the queue. 
    """ 
    result = _op_def_lib.apply_op("FIFOQueueV2", 
           component_types=component_types, 
           shapes=shapes, capacity=capacity, 
           container=container, shared_name=shared_name, 
           name=name) 
    return result 

如果您熟悉了这种依赖，你一定队列赢得不会吃掉所有可用内存，您可以通过-1，从而大大简化您的代码。