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/api_docs/python/tf/compat/v1/train/maybe_batch