在多线程Tensorflow中重新分发批处理结果

Question

我正在使用

对TF.FIFOqueue的数据进行多次输入

q = tf.FIFOQueue(...)

数据由线程生成，每个线程使用q.enqueue加载数据点，并等待评估。批处理准备就绪后，我使用

获取输出

batch = q.dequeue_many(batch_size)

使用一些NN处理数据。 现在，这是我的问题： 如何将结果重新分配给每个线程，以便每个线程都能获得其数据点的预测？换句话说，我如何知道哪个线程放置了条目＃1，以便用输出＃1提供它？

或者，我应该以不同的方式处理问题吗？

Answer 1

您可以将张量的元组排入队列/从队列中排队。一种简单的方法是将一对(thread_id, data_point)排队，而不只是data_point。例如，thread_id可以是常数整数或字符串张量。