tf.compat.v1.train.input_producer
Выводит строки из input_tensor в очередь для входного потока. (устарело)
tf.compat.v1.train.input_producer(
input_tensor,
element_shape=None,
num_epochs=None,
shuffle=True,
seed=None,
capacity=32,
shared_name=None,
summary_name=None,
name=None,
cancel_op=None
)
Примечание: еслиnum_epochsнеNone, эта функция создает локальный счётчикepochs. Используйтеlocal_variables_initializer()для инициализации локальных переменных.
| Аргументы | |
|---|---|
input_tensor | Массив со строками, которые нужно сгенерировать. Должен быть одномерным или иметь определённую форму, или element_shape должна быть определена. |
element_shape | (Необязательно.) TensorShape представляющий форму строки из input_tensor, если её невозможно вывести. |
num_epochs | (Необязательно.) Целое число. Если указано, input_producer производит каждую строку из input_tensor num_epochs раз, прежде чем генерировать ошибку OutOfRange. Если не указано, input_producer может циклически перебирать строки из input_tensor неограниченное количество раз. |
shuffle | (Необязательно.) Булево значение. Если true, строки перемешиваются случайным образом в каждом эпохе. |
seed | (Необязательно.) Целое число. Семена для использования, если shuffle имеет значение true. |
capacity | (Необязательно.) Ёмкость очереди для буферизации ввода. |
shared_name | (Необязательно.) Если задано, эта очередь будет использоваться совместно под указанным именем в нескольких сессиях. |
summary_name | (Необязательно.) Если задано, скалярный отчёт о текущем размере очереди будет сгенерирован, используя это имя как часть тега. |
name | (Необязательно.) Имя очереди. |
cancel_op | (Необязательно.) Операция отмены для очереди |
| Возвращает | |
|---|---|
Очередь с выходными строками. QueueRunner для очереди добавляется в текущую QUEUE_RUNNER коллекцию текущей схемы. |
| Исключения | |
|---|---|
ValueError | Если форму входных данных невозможно вывести из аргументов. |
RuntimeError | Если вызывается с включённым немедленным выполнением. |
совместимость с eager
Потоки ввода, основанные на очередях, не поддерживаются при включённом немедленном выполнении. Пожалуйста, используйте API tf.data для ввода данных при немедленном выполнении.
© 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/train/input_producer