python - 如何使用 tf.data 数据集在同一批处理上执行多个训练操作

Question

我想实现可用的代码 here使用 tf.data 数据集和迭代器(对抗性自动编码器)。

我的问题是如何使用同一批进行多个训练操作？

我需要在同一批处理上执行三个不同的训练操作，但是，如果我使用 tf.data 迭代器，它不适用于同一批处理，而是适用于连续批处理。

Answer 1

假设 data = db_iter.get_next()，其中 db_iter 是您在 tf.data 中使用的迭代器。

我想您正在 3 不同的 sess.run 语句中训练 3 操作。在这种情况下，它们都将使用 3 不同批处理，因为 data 将被评估 3 次。

事实是，如果三个训练操作中每一个的输入均由上面定义的data提供，并在每个sess.run中运行，他们都将使用同一批处理。