tf.raw_ops.PartitionedCall

возвращает f(inputs), где размещается и разделяется тело f.

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

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

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

tf.compat.v1.raw_ops.PartitionedCall

tf.raw_ops.PartitionedCall(
    args,
    Tout,
    f,
    config='',
    config_proto='',
    executor_type='',
    name=None
)

Асинхронно выполняет функцию, потенциально на нескольких устройствах, но в рамках одного процесса. Ядро размещает и разбиение подлежащего графа данной функции и выполняет каждый из разделенных подграфов как функцию.

Аргументы
`args`	Список объектов `Tensor`. Список входных тензоров.
`Tout`	Список `tf.DTypes`. Список типов вывода.
`f`	Функция, декорированная @Defun. Функция, которая принимает 'args', список тензоров, и возвращает 'output', другой список тензоров. Типы ввода и вывода задаются 'Tin' и 'Tout'. Тело функции f будет размещено и разделено по устройствам, что отличает этот оператор от обычного оператора Call.
`config`	Необязательный `string`. По умолчанию `""`.
`config_proto`	Необязательный `string`. По умолчанию `""`.
`executor_type`	Необязательный `string`. По умолчанию `""`.
`name`	Имя операции (необязательно).

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

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