tf.types.experimental.AtomicFunction

Базовый класс графовых функций.

Наследует: Callable

AtomicFunction инкапсулирует определение одной графовой функции.

AtomicFunction можно вызывать напрямую, только если согласно FunctionType не требуется захватов. Если захваты есть, используйте вместо этого call_with_captures.

AtomicFunction не поддерживает градиенты. Если вам нужна поддержка градиентов, используйте родительский класс ConcreteFunction.

Методы

call_with_captures

Просмотреть исходный код

call_with_captures(
    args, kwargs, captures
)

Вызывает эту AtomicFunction с захватами, определёнными её FunctionType.

__call__

Просмотреть исходный код

__call__(
    *args, **kwargs
)

Выполняет этот вызываемый объект.

Работает как обычная операция: в eager-режиме выполнение начинается немедленно и возвращаются результаты. В режиме графа создаются операции, возвращающие символические значения TensorFlow (например, tf.Tensor, tf.data.Dataset и т. д.). Например, вызываемые объекты tf.function обычно создают операцию tf.raw_ops.PartitionedCall, но не всегда: конкретный набор создаваемых операций является деталью внутренней реализации.