如何在Tensorflow V2中将tf.data.Dataset与图表一起使用?

时间:2019-03-06 00:38:58

标签: python tensorflow tensorflow-datasets

我发现Tensorflow V2中已弃用了诸如make_one_shot_iterator之类的功能,建议使用for ... in dataset,这需要启用急切的执行。我很困惑没有 这些功能,如何使用图形从Dataset提取数据?在我看来,唯一的选择就是转到tf.compat.v1.data.make_one_shot_iterator(dataset)。 V2中有建议的选择吗?为什么要删除这些功能?

1 个答案:

答案 0 :(得分:1)

根据官方文档,数据集在TF 2.0中将变为可迭代,并且默认模式下将启用急切模式。您可以在这里找到更多有关它的信息
 https://github.com/tensorflow/docs/blob/master/site/en/r2/guide/effective_tf2.md