tf.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, name=None
)
Создаёт набор данных, который применяет f к результатам input_dataset и затем группирует batch_size из них в пакеты.
В отличие от "MapDataset", который применяет f последовательно, этот набор данных вызывает до batch_size * num_parallel_batches копий f параллельно.
| Аргументы | |
|---|---|
input_dataset |
Tensor типа variant. Значение типа 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/MapAndBatchDataset