random_batch 如何在 Tensorflow 中弃用?

How is random_batch deprecated in Tensorflow?

我正在做 this 教程,但我卡在了 data.random_batch(batch_size=train_batch_size) 的那一行。 看起来张量流中有一些弃用。我收到以下错误:

AttributeError: 'Datasets' object has no attribute 'random_batch'

我一直在谷歌上搜索解决方案,但没有成功。有谁知道这个方法现在隐藏在哪里?

您可以使用 tf.data.dataset.batch(batch_size = train_batch_size) 对输入数据进行批处理,但首先您必须使用数据的相关方法从输入数据创建数据集,例如 dataset = tf.data.TFRecordDataset(filename)。 之后,您可以创建一个迭代器,通过定义一个迭代器 dataset.make_one_shot_iterator() 来获取每个批次进行训练。可以在 tensorflow 指南中找到详细的解释 here