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`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/ExperimentalMapAndBatchDataset