tf.raw_ops.MapDefun

Применяет функцию к списку тензоров, распакованных из аргументов по размерности 0.

Псевдонимы

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

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

tf.compat.v1.raw_ops.MapDefun

tf.raw_ops.MapDefun(
    arguments,
    captured_inputs,
    output_types,
    output_shapes,
    f,
    max_intra_op_parallelism=1,
    name=None
)

Функция, заданная значением f, предполагается бессостоятельной и выполняется параллельно на всех срезах; до batch_size (то есть размера 0-й размерности каждого аргумента) функций будет запланировано одновременно.

Атрибут max_intra_op_parallelism, который по умолчанию равен 1, можно использовать для ограничения внутриоператорного параллелизма. Для ограничения меж-операторного параллелизма пользователь может установить частный пул потоков на наборе данных, используя tf.data.Options's ThreadingOptions.

Обратите внимание, что этот оператор не предоставляется пользователям напрямую, а вызывается в переписывании tf.data.

Аргументы
`arguments`	Список объектов `Tensor`. Список тензоров, типы которых `Targuments`, соответствующие входным данным, над которыми должна быть применена функция.
`captured_inputs`	Список объектов `Tensor`. Список тензоров, типы которых `Tcaptured`, соответствующие захваченным входным данным defun.
`output_types`	Список `tf.DTypes`, длина которого равна `>= 1`. Список типов.
`output_shapes`	Список форм (каждая — `tf.TensorShape` или список `ints`), длина которого равна `>= 1`. Список форм.
`f`	Функция, декорированная @Defun.
`max_intra_op_parallelism`	Необязательный `int`. По умолчанию `1`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Список объектов `Tensor` типа `output_types`.

© 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/MapDefun