tf.compat.v1.train.string_input_producer
Вывод строк (например, имён файлов) в очередь для входной обработки. (устарело)
tf.compat.v1.train.string_input_producer(
string_tensor, num_epochs=None, shuffle=True, seed=None, capacity=32,
shared_name=None, name=None, cancel_op=None
)
Примечание: еслиnum_epochsнеNone, эта функция создаёт локальный счётчикepochs. Используйтеlocal_variables_initializer()для инициализации локальных переменных.
| Аргументы | |
|---|---|
string_tensor | 1-мерный тензор строк со строками для производства. |
num_epochs | Целое число (необязательно). Если указано, string_input_producer производит каждую строку из string_tensor num_epochs раз перед генерацией ошибки OutOfRange. Если не указано, string_input_producer может циклически проходить через строки string_tensor неограниченное количество раз. |
shuffle | Булево. Если True, строки случайным образом перемешиваются в рамках каждой эпохи. |
seed | Целое число (необязательно). Зерно, используемое, если shuffle == True. |
capacity | Целое число. Устанавливает ёмкость очереди. |
shared_name | (необязательно). Если установлено, эта очередь будет использоваться совместно под заданным именем в нескольких сессиях. Все сессии, открытые для устройства, имеющего эту очередь, смогут получить к ней доступ через shared_name. Использование этого в распределённой среде означает, что каждое имя будет видно только одной из сессий, имеющей доступ к этой операции. |
name | Имя для операций (необязательно). |
cancel_op | Операция отмены для очереди (необязательно). |
| Возвращает | |
|---|---|
Очередь с выходными строками. A QueueRunner для очереди добавляется в текущую Graph коллекцию QUEUE_RUNNER. |
| Возможные исключения | |
|---|---|
ValueError | Если string_tensor является пустым списком Python. Во время выполнения произойдёт ошибка с утверждением, если string_tensor станет нулевым тензором. |
Совместимость с Eager
Входные конвейеры на основе очередей не поддерживаются при включённом eager выполнении. Пожалуйста, используйте API tf.data для ввода данных при eager выполнении.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/train/string_input_producer