python - 如何从 tensorflow 数据集迭代器返回同一批处理两次？

Question

我正在将一些遗留代码转换为使用数据集 API - 此代码使用 feed_dict 将一批提供给训练操作(实际上是三次)，然后重新计算损失以供显示 使用同一批。所以我需要一个迭代器，它可以两次(或多次)返回完全相同的批处理。不幸的是，我似乎无法找到一种方法来使用 tensorflow 数据集 - 这可能吗？

Answer 1

您可以使用 Dataset.flat_map() 重复 Dataset 的单个元素, Dataset.from_tensors()和 Dataset.repeat()一起。例如，重复元素两次:

NUM_REPEATS = 2
dataset = tf.data.Dataset.range(10)  # ...or the output of `.batch()`, etc.

# Repeat each element of `dataset` NUM_REPEATS times.
dataset = dataset.flat_map(
    lambda x: tf.data.Dataset.from_tensors(x).repeat(NUM_REPEATS))