tf.train.maybe_batch
Условно создаёт пакеты тензоров на основе keep_input. (устарело)
tf.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/r1.15/api_docs/python/tf/train/maybe_batch