tf.compat.v1.train.shuffle_batch

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

tf.compat.v1.train.shuffle_batch(
    tensors, batch_size, capacity, min_after_dequeue, num_threads=1, seed=None,
    enqueue_many=False, shapes=None, allow_smaller_final_batch=False,
    shared_name=None, name=None
)

Эта функция добавляет следующее к текущему Graph:

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

Если 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, если входная очередь исчерпана. Если эта операция подаёт данные в другую очередь ввода, её очередь выполнения поймает эту ошибку, однако, если эта операция используется в вашем основном потоке, вы должны поймать её сами.

Например:

# Creates batches of 32 images and 32 labels.
image_batch, label_batch = tf.compat.v1.train.shuffle_batch(
      [single_image, single_label],
      batch_size=32,
      num_threads=4,
      capacity=50000,
      min_after_dequeue=10000)

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

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

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

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