tl;dr; 我可以限制 tf.data.Dataset 中的元素数量吗？
A 有一个训练和评估循环来处理整个给定的数据集。这对于测试来说并不理想，因为它需要很长时间才能遍历整个数据集。我可以通过创建一个 Mock 数据集或通过限制 dataset 的元素数量来测试此代码，这样代码仅通过，比方说，前 10 个数据点。我怎样才能做第二个？
谢谢
最佳答案
从 Dataset 中仅获取固定数量的元素 n 的最简单方法是使用 Dataset.take(n)。例如:
large_dataset = ...
small_dataset = large_dataset.take(10)
关于testing - 限制 tf.data.Dataset 中的项目数，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/51006470/