Python 带有tf.data.Dataset的RandomShuffleQueue功能

我想用

tf.data.Dataset

替换我以前的

randomshuffleequeue

方法。对于一些背景：我在运行时生成数据，将其放入队列，然后随机取出

我看不到用

tf.data.Dataset

实现这一点的方法，因为我总是需要生成数据，而我没有办法将新数据放入

Dataset

。像repeat这样的东西对我没有帮助，因为它总是在

数据集

上使用我预先放置的相同元素

---

你知道如何将

队列

功能放入

tf.data.Dataset

？基本上，如果我在

tf.data.DataSet

中有一个队列/缓冲区，我如何用新数据重新填充它？

如果我理解正确，这听起来像是一个完美的匹配。如果您想创建一个缓冲区并从中随机提取元素，您可以在之后添加

Dataset.shuffle（）

。

是的，这看起来真的很不错，谢谢。缺少的内容类似于数据集将占用的元素的边界，类似于您可以为队列中的最大元素设置边界。您是否查看了

DataSet.take（count）

？听起来很相似。