Spec-Zone.ru › TensorFlow 2.3

tf.raw_ops.ExperimentalMapAndBatchDataset

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

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

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

Дополнительные сведения см. в Руководстве по миграции.

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

Spec-Zone.ru

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