tf.compat.v1.data.make_initializable_iterator
Создаёт итератор для элементов dataset.
tf.compat.v1.data.make_initializable_iterator(
dataset, shared_name=None
)
Миграция на TF2
Это устаревший 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)
| Аргументы | |
|---|---|
dataset | A tf.data.Dataset. |
shared_name | (Необязательно.) Если не пусто, возвращаемый итератор будет совместно использоваться под данным именем в нескольких сессиях, которые используют одни и те же устройства (например, при использовании удалённого сервера). |
| Возвращаемое значение | |
|---|---|
A tf.data.Iterator для элементов dataset. |
| Возможные исключения | |
|---|---|
RuntimeError | Если включено выполнение в режиме eager. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/data/make_initializable_iterator