tf.compat.v1.train.shuffle_batch_join

Создать пакеты, случайным образом перемешивая тензоры. (устарело)

tf.compat.v1.train.shuffle_batch_join(
    tensors_list, batch_size, capacity, min_after_dequeue, seed=None,
    enqueue_many=False, shapes=None, allow_smaller_final_batch=False,
    shared_name=None, name=None
)

Аргумент tensors_list представляет собой список кортежей тензоров или список словарей тензоров. Каждый элемент в списке обрабатывается аналогично аргументу tensors функции tf.compat.v1.train.shuffle_batch().

Эта версия помещает в очередь различные списки тензоров в разных потоках. Она добавляет следующее к текущему Graph:

• Очередь перемешивания, в которую помещаются тензоры из tensors_list.
• Операцию dequeue_many для создания пакетов из очереди.
• Операцию QueueRunner в коллекцию QUEUE_RUNNER, для помещения тензоров из tensors_list в очередь.

Будет запущено len(tensors_list) потоков, при этом поток i помещает в очередь тензоры из tensors_list[i]. tensors_list[i1][j] должны соответствовать tensors_list[i2][j] по типу и форме, за исключением первого измерения, если enqueue_many равно true.

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

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

Аргумент capacity управляет тем, как долго предварительная выборка может увеличивать очереди.

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

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

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

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