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

FIFO队列在队列分别为满或空时,会在其enqueuedequeue函数中暂停。如果只有一个队列,这不是问题。

我的情况是有两个队列,我需要根据以下条件进行入队操作:

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_openqueue_diff_op分别是queue_samequeue_diff的操作。

由于l1<>l2的概率远大于l1==l2,因此即使我采用了run_options,会话也会在queue_diff的入队操作上暂停。

有没有办法获取队列的大小,或者测试队列是否已满?

非常感谢您的帮助。


回答:

有没有办法获取队列的大小,或者测试队列是否已满?

您可以使用size()函数来设置队列的大小,并与队列容量进行比较

capacity=1000 # 您的队列容量dtype=tf.float32queue_diff = tf.FIFOQueue(capacity, dtype)enqueue_op = tf.cond(tf.equal(queue_diff.size(), capacity), lambda:is_full(), lambda: enqueue(queue_diff, enqueue_elements..))# 当队列已满时处理的函数def is_full():    return 'Enqueue_Failed'# 入队操作的函数def enqueue(queue, element...):    queue.enqueue(element)    return 'Enqueue_Success'

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注