在tensorflow中运行Iterator.get_next()后如何获取上一批？

Question

最近想实现 GAN 模型，使用 tf.Dataset 和 Iterator 读取人脸图像作为训练数据。

数据集和迭代器对象的代码是：

self.dataset = tf.data.Dataset.from_tensor_slices(convert_to_tensor(self.data_ob.train_data_list, dtype=tf.string))
self.dataset = self.dataset.map(self._parse_function)
#self.dataset = self.dataset.shuffle(buffer_size=10000)
self.dataset = self.dataset.apply(tf.contrib.data.batch_and_drop_remainder(batch_size))

self.iterator = tf.data.Iterator.from_structure(self.dataset.output_types, self.dataset.output_shapes)
self.next_x = self.iterator.get_next()

我的新 GAN 模型是：

self.z_mean, self.z_sigm = self.Encode(self.next_x)
self.z_x = tf.add(self.z_mean, tf.sqrt(tf.exp(self.z_sigm))*self.ep)
self.x_tilde = self.generate(self.z_x, reuse=False)
#the feature
self.l_x_tilde, self.De_pro_tilde = self.discriminate(self.x_tilde)

#for Gan generator
self.x_p = self.generate(self.zp, reuse=True)
# the loss of dis network
self.l_x,  self.D_pro_logits = self.discriminate(self.next_x, True)

所以，问题是我两次使用 self.next_x 作为输入张量。每次的数据集都不一样。那么，如何解决这个问题以保留第一批以供重复使用呢？

Answer 1

我在代码中使用的内容如下，其中 x 和 y_true 是占位符。不确定是否有更有效的实现。

images, labels = session.run(next_element)
batch_accuracy = session.run(accuracy, feed_dict={x: images, y_true: labels, keep_prop: 1.0})
batch_predicted_probabilities = session.run(y_pred, feed_dict={x: images, y_true: labels, keep_prop: 1.0})

我目前正在尝试使用 tf.placeholder_with_default 而不是 x 和 y_true 的普通占位符来检查它是否在我的项目中提供了更好的性能。如果我能很快得到任何结果，我会编辑我的答案让你知道:)。

编辑： 我切换到 placeholder_with_default 并没有显着提高每批次的速度，至少在我测量它的方式上。