具有tf.data.Dataset的RandomShuffleQueue功能

Question

我想代替我的老RandomShuffleQueue与方法tf.data.Dataset 。 对于某些背景：我在运行时生成数据，将其放入队列中，然后将其随机取出。

我看不到使用tf.data.Dataset做到这一点的tf.data.Dataset ，因为我总是需要生成数据，而我没有办法将新数据放入Dataset 。 诸如repeat之类的东西不会对我有帮助，因为仅在Dataset使用与我之前放入的相同的元素就可以正常工作。

任何想法如何将Queue功能添加到tf.data.Dataset ？ 基本上，如果我在tf.data.DataSet有一个队列/缓冲区，如何用新数据重新填充它？

Answer 1

如果我对您的理解正确，这听起来像是Dataset.from_generator()的完美匹配。 如果要创建缓冲区并从中随机取出元素，则可以随后添加Dataset.shuffle() 。