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 . |
© 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.3/api_docs/python/tf/raw_ops/ExperimentalMapAndBatchDataset