tf.compat.v1.train.maybe_shuffle_batch
Создаёт пакеты, случайным образом перемешивая условно добавляемые в очередь тензоры. (устарела)
tf.compat.v1.train.maybe_shuffle_batch(
tensors,
batch_size,
capacity,
min_after_dequeue,
keep_input,
num_threads=1,
seed=None,
enqueue_many=False,
shapes=None,
allow_smaller_final_batch=False,
shared_name=None,
name=None
)
Подробнее см. docstring в shuffle_batch.
| Аргументы | |
|---|---|
tensors | Список или словарь тензоров для добавления в очередь. |
batch_size | Новый размер пакета, извлекаемого из очереди. |
capacity | Целое число. Максимальное количество элементов в очереди. |
min_after_dequeue | Минимальное количество элементов в очереди после извлечения, необходимое для обеспечения определённого уровня перемешивания элементов. |
keep_input | Тензор bool. Этот тензор определяет, добавлять ли входные данные в очередь. Если это скаляр со значением True, то все tensors добавляются в очередь. Если это вектор и enqueue_many имеет значение True, то каждый пример добавляется в очередь, только если соответствующее значение в keep_input равно True. По сути, этот тензор служит механизмом фильтрации. |
num_threads | Количество потоков, добавляющих tensor_list в очередь. |
seed | Начальное значение для случайного перемешивания в очереди. |
enqueue_many | Является ли каждый тензор в tensor_list отдельным примером. |
shapes | (Необязательно) Формы каждого примера. По умолчанию используются формы, выведенные для tensor_list. |
allow_smaller_final_batch | (Необязательно) Логическое значение. Если True, разрешить уменьшенный размер последнего пакета, если в очереди осталось недостаточно элементов. |
shared_name | (Необязательно) Если задано, эта очередь будет использоваться совместно в нескольких сеансах под указанным именем. |
name | (Необязательно) Имя операций. |
| Возвращает | |
|---|---|
Список или словарь тензоров с типами, соответствующими tensors. |
| Вызывает исключения | |
|---|---|
ValueError | Если shapes не указаны и их невозможно вывести из элементов tensors. |
совместимость с eager execution
Конвейеры ввода на основе очередей не поддерживаются при включённом eager execution. Для загрузки данных в режиме eager execution используйте API tf.data.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/train/maybe_shuffle_batch