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. Векторный тип, представляющий входной набор данных. |
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/MapAndBatchDataset
