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.

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

Если 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, будут иметь значение None для первой Dimension, а операции, зависящие от фиксированного batch_size, завершатся с ошибкой.

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