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