具有tf.data.Dataset的RandomShuffleQueue功能

Question

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

我没有办法用tf.data.Dataset来做到这一点，因为我总是需要生成数据，而我没有办法将新数据放入Dataset中。诸如重复之类的东西对我没有帮助，因为仅在Dataset上使用与我事先放置的元素相同的元素即可。

有什么想法可以将Queue功能引入tf.data.Dataset？基本上，如果我在tf.data.DataSet中有一个队列/缓冲区，如何用新数据重新填充它？

Answer 1

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