Python tf.compat.v1.data.make_initializable_iterator用法及代碼示例

為 dataset 的元素創建一個迭代器。

用法

tf.compat.v1.data.make_initializable_iterator(
    dataset, shared_name=None
)

遷移到 TF2

警告：這個 API 是為 TensorFlow v1 設計的。繼續閱讀有關如何從該 API 遷移到本機 TensorFlow v2 等效項的詳細信息。見TensorFlow v1 到 TensorFlow v2 遷移指南有關如何遷移其餘代碼的說明。

這是用於使用數據集元素的舊 API，僅應在從 TF 1 過渡到 TF 2 期間使用。請注意，使用此 API 應該是代碼庫的臨時狀態，因為通常不能保證 TF 1 的互操作性和 TF 2 代碼。

在 TF 2 中，數據集是 Python 可迭代對象，這意味著您可以使用 for elem in dataset:... 或通過顯式創建迭代器 iterator = iter(dataset) 並通過 values = next(iterator) 獲取其元素來使用它們的元素。

注意：返回的迭代器將處於未初始化狀態，您必須在使用前運行iterator.initializer 操作：

dataset = ...
iterator = tf.compat.v1.data.make_initializable_iterator(dataset)
# ...
sess.run(iterator.initializer)