tf.raw_ops.MapAndBatchDataset

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

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

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

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

tf.compat.v1.raw_ops.MapAndBatchDataset

tf.raw_ops.MapAndBatchDataset(
    input_dataset,
    other_arguments,
    batch_size,
    num_parallel_calls,
    drop_remainder,
    f,
    output_types,
    output_shapes,
    preserve_cardinality=False,
    metadata='',
    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`.
`metadata`	Необязательный `string`. Значение по умолчанию — `""`.
`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/MapAndBatchDataset