tf.compat.v1.train.batch

Создаёт пакеты тензоров в tensors. (устарело)

tf.compat.v1.train.batch(
    tensors, batch_size, num_threads=1, capacity=32, enqueue_many=False,
    shapes=None, dynamic_pad=False, allow_smaller_final_batch=False,
    shared_name=None, name=None
)

Аргумент tensors может быть списком или словарем тензоров. Значение, возвращаемое функцией, будет того же типа, что и tensors.

Эта функция реализована с помощью очереди. QueueRunner для очереди добавляется в текущую Graph's QUEUE_RUNNER коллекцию.

Если enqueue_many равно False, предполагается, что tensors представляет собой отдельный пример. Входной тензор с формой [x, y, z] будет выводиться как тензор с формой [batch_size, x, y, z].

Если enqueue_many равно True, предполагается, что tensors представляет пакет примеров, где первый размер индексируется по примеру, а все члены tensors должны иметь одинаковый размер в первом измерении. Если входной тензор имеет форму [*, x, y, z], вывод будет иметь форму [batch_size, x, y, z]. Аргумент capacity управляет тем, как долго предварительная выборка может увеличивать очереди.

Возвращаемая операция — это операция декьюирования и выбросит tf.errors.OutOfRangeError, если входная очередь исчерпана. Если эта операция подключается к другой очереди ввода, её исполнитель очереди перехватит эту ошибку. Однако, если эта операция используется в вашем основном потоке, вы сами несёте ответственность за перехват этой ошибки.

Примечание: Если dynamic_pad равно False, вы должны убедиться, что либо (i) передан аргумент shapes, либо (ii) все тензоры в tensors должны иметь полностью определённые формы. ValueError будет поднято, если ни одно из этих условий не выполняется.

Если dynamic_pad равно True, достаточно знать ранг тензоров, но отдельные размеры могут иметь форму None. В этом случае для каждого добавления в очередь размеры со значением None могут иметь переменную длину; при извлечении из очереди выходные тензоры будут дополнены справа до максимальной формы тензоров в текущем мини-пакете. Для чисел это дополнение принимает значение 0. Для строк это пустая строка. Смотрите PaddingFIFOQueue для более подробной информации.

Если allow_smaller_final_batch равно True, возвращается меньшее значение размера пакета, чем batch_size, когда очередь закрывается и нет достаточно элементов, чтобы заполнить пакет; в противном случае ожидающие элементы отбрасываются. Кроме того, все статические формы выходных тензоров, доступные через свойство shape, будут иметь первое значение Dimension равное None, и операции, зависящие от фиксированного размера пакета, завершатся ошибкой.

Совместимость с Eager

Конвейеры ввода данных, основанные на очередях, не поддерживаются при включенном режиме eager. Пожалуйста, используйте API tf.data для обработки данных в режиме eager.