Spec-Zone.ru › TensorFlow

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 A Tensor типа variant. Вариантный тензор, представляющий входной набор данных.
other_arguments Список объектов Tensor. Список тензоров, обычно значения, которые были захвачены при построении замыкания для f.
batch_size A Tensor типа int64. Скаляр, представляющий количество элементов, которые следует накапливать в пакете. Он определяет количество одновременных вызовов f, которые обрабатывают элементы из input_dataset параллельно.
num_parallel_calls A Tensor типа int64. Скаляр, представляющий максимальное количество одновременных вызовов функции map_fn. Применение map_fn к последовательным элементам ввода параллельно может улучшить пропускную способность канала ввода.
drop_remainder A Tensor типа bool. Скаляр, представляющий, следует ли пропускать последний пакет в случае, если его размер меньше желаемого.
f Функция, оформленная как @Defun. Функция для применения к результатам input_dataset.
output_types Список tf.DTypes длиной >= 1.
output_shapes Список форм (каждая — tf.TensorShape или список ints), длина которого равна >= 1.
preserve_cardinality Необязательный bool. По умолчанию False.
name Имя операции (необязательно).
Возвращаемое значение
A Tensor типа variant.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/ExperimentalMapAndBatchDataset

Spec-Zone.ru

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