限制 tf.data.Dataset 中的项目数
Limiting the number of items in a tf.data.Dataset
tl;dr; 我可以限制 tf.data.Dataset 中的元素数量吗？
A 有一个处理整个给定 dataset 的训练和评估循环。这对于测试来说并不理想，因为它需要很长时间才能遍历整个数据集。我可以通过创建 Mock 数据集或通过限制 dataset 的元素数量来测试此代码，因此代码仅通过，比方说，前 10 个数据点。第二个怎么办？
谢谢
从 Dataset 中仅获取固定数量的元素 n 的最简单方法是使用 Dataset.take(n)。例如：
large_dataset = ...
small_dataset = large_dataset.take(10)
tl;dr; 我可以限制 tf.data.Dataset 中的元素数量吗？
A 有一个处理整个给定 dataset 的训练和评估循环。这对于测试来说并不理想，因为它需要很长时间才能遍历整个数据集。我可以通过创建 Mock 数据集或通过限制 dataset 的元素数量来测试此代码，因此代码仅通过，比方说，前 10 个数据点。第二个怎么办？
谢谢
从 Dataset 中仅获取固定数量的元素 n 的最简单方法是使用 Dataset.take(n)。例如：
large_dataset = ...
small_dataset = large_dataset.take(10)