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 в коллекцию 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.