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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/train/maybe_batch