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
)
Дополнительные сведения см. в docstring в 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/api_docs/python/tf/compat/v1/train/maybe_batch