tf.types.experimental.AtomicFunction

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

Наследуется от: Callable

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

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, но не всегда — точные генерируемые операции являются внутренней реализацией.