Spec-Zone.ru › TensorFlow 2.4

tf.raw_ops.MapAndBatchDataset

Создаёт набор данных, который объединяет применение функции к элементам с формированием пакета.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API