如何使用 tf.data 数据集对同一批次执行多个训练操作

Question

我想使用 tf.data 数据集和迭代器（对抗性自动编码器）来实现这里可用的代码。我的问题是如何将同一批次用于多个训练操作？我需要在同一个批次上执行三个不同的训练操作，但是，如果我使用 tf.data 迭代器，它不会在同一个批次上工作，而是在连续的批次上工作。