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`	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`.
`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/versions/r2.9/api_docs/python/tf/raw_ops/ExperimentalMapAndBatchDataset