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 execution. |
Совместимость с eager execution
Конвейеры ввода на основе очередей не поддерживаются при включенном eager execution. Для загрузки данных при eager execution используйте 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/api_docs/python/tf/compat/v1/train/input_producer