Spec-Zone.ru › TensorFlow 2.3

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.

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

Spec-Zone.ru

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