tf.compat.v1.train.maybe_batch
Условно создаёт пакеты тензоров на основе keep_input. (устарело)
tf.compat.v1.train.maybe_batch(
tensors,
keep_input,
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
)
См. строку документации в batch для получения более подробной информации.
| Аргументы | |
|---|---|
tensors | Список или словарь тензоров для очереди. |
keep_input | Тензор bool. Этот тензор контролирует, добавляется ли вход в очередь или нет. Если это скаляр, и он оценивается True, то tensors добавляются в очередь. Если это вектор, и enqueue_many имеет значение True, то каждый пример добавляется в очередь только если соответствующее значение в keep_input равно True. Этот тензор по существу действует как механизм фильтрации. |
batch_size | Новый размер пакета, извлечённый из очереди. |
num_threads | Количество потоков, добавляющих tensors. Операция пакетной обработки будет недетерминированной, если num_threads > 1. |
capacity | Целое число. Максимальное количество элементов в очереди. |
enqueue_many | Является ли каждый тензор в tensors отдельным примером. |
shapes | (Необязательно) Форматы для каждого примера. По умолчанию это предполагаемые форматы для tensors. |
dynamic_pad | Булево значение. Разрешить переменные размеры в форматах входных данных. При извлечении данные заполняются для того, чтобы тензоры в пакете имели одинаковые форматы. |
allow_smaller_final_batch | (Необязательно) Булево значение. Если True, разрешить конечный пакет меньшего размера, если в очереди недостаточно элементов. |
shared_name | (Необязательно). Если задано, эта очередь будет совместно использоваться под заданным именем в нескольких сеансах. |
name | (Необязательно) Имя для операций. |
| Возвращает | |
|---|---|
Список или словарь тензоров с теми же типами, что и tensors. |
| Возможные исключения | |
|---|---|
ValueError | Если shapes не указаны и не могут быть выведены из элементов tensors. |
© 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/versions/r2.9/api_docs/python/tf/compat/v1/train/maybe_batch