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/versions/r2.9/api_docs/python/tf/raw_ops/MapDefun