从Iterator(从tf.data.Dataset)向TensorFlow模型中馈送元素的有效方法是什么?

问题描述 投票:3回答:1

我正在使用TensrFlow的新API通过tf.data.Dataset和迭代器导入数据。它工作正常,但是我不确定我的工作是否有效。

[我目前正在做的是评估迭代器的get_next()方法,该方法为我提供了一堆元素,例如实际图像,其标签,文件名等。然后,我使用[C0 ]。

[我知道feed_dict非常慢,所以我会失去feed_dict和Iterators的好处,并通过评估条目并将它们通过Dataset馈入图形而在TFRecord中拥有序列化的数据集吗?我在TF的文档中找不到任何示例,该示例说明了人们期望如何使用Iterator的feed_dict将元素馈入模型。最好解压缩get_next()并直接在图形中使用结果?

tensorflow tensorflow-datasets
1个回答
1
投票

是,建议直接在图形中使用get_next()返回的张量。也许您错过了get_next() API上的Programmer's Guide

您可能还会发现tf.data说明性的。

希望有所帮助。

© www.soinside.com 2019 - 2024. All rights reserved.