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