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

Например:

# 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, и операции, зависящие от фиксированного batch_size, не будут работать.

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