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

Возвращаемые значения
A `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