Spec-Zone.ru › TensorFlow

tf.raw_ops.ExperimentalMapAndBatchDataset

Создает набор данных, объединяющий отображение и пакетную обработку.

Просмотреть псевдонимы

Совместимые псевдонимы для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.ExperimentalMapAndBatchDataset

tf.raw_ops.ExperimentalMapAndBatchDataset(
    input_dataset,
    other_arguments,
    batch_size,
    num_parallel_calls,
    drop_remainder,
    f,
    output_types,
    output_shapes,
    preserve_cardinality=False,
    name=None
)

Создает набор данных, который применяет f к результатам input_dataset, а затем объединяет batch_size в пакеты.

В отличие от «MapDataset», который последовательно применяет f, этот набор данных вызывает до batch_size * num_parallel_batches копий f параллельно.

Аргументы
input_dataset Tensor типа variant. Тензор варианта, представляющий входной набор данных.
other_arguments Список объектов Tensor. Список тензоров, обычно значений, захваченных при создании замыкания для f.
batch_size Tensor типа int64. Скаляр, задающий количество элементов, накапливаемых в пакете. Он определяет количество параллельных вызовов f, обрабатывающих элементы из input_dataset.
num_parallel_calls Tensor типа int64. Скаляр, задающий максимальное количество параллельных вызовов функции map_fn. Применение map_fn к последовательным входным элементам параллельно может повысить пропускную способность входного конвейера.
drop_remainder Tensor типа bool. Скаляр, указывающий, следует ли отбрасывать последний пакет, если его размер меньше требуемого.
f Функция, декорированная с помощью @Defun. Функция, применяемая к результатам input_dataset.
output_types Список tf.DTypes длиной >= 1.
output_shapes Список форм (каждая из них — tf.TensorShape или список ints) длиной >= 1.
preserve_cardinality Необязательный bool. По умолчанию — False.
name Имя операции (необязательно).
Возвращает
Tensor типа variant.

© 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/raw_ops/ExperimentalMapAndBatchDataset

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API