tf.data.Dataset的RandomShuffleQueue功能

Question

我想用tf.data.Dataset替换我的旧RandomShuffleQueue方法。对于一些背景知识:我在运行时生成数据，将其放入队列中，然后随机取出。

我看不到使用tf.data.Dataset的方法，因为我总是需要生成数据，而我又没有办法将新数据放入Dataset中。像repeat这样的东西对我没有帮助，因为它总是使用我预先设置的相同元素在Dataset上工作。

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

Answer 1

如果我没理解错的话，它听起来很适合Dataset.from_generator()。如果您想创建一个缓冲区并从中随机提取元素，您可以在之后添加Dataset.shuffle()。