tf.types.experimental.SupportsTracingProtocol

Протокол, позволяющий настраивать перетрассировку tf.function для пользовательских классов.

tf.types.experimental.SupportsTracingProtocol(
    *args, **kwargs
)

Методы

__tf_tracing_type__

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

@abc.abstractmethod
__tf_tracing_type__(
    context: tf.types.experimental.TracingContext
) -> tf.types.experimental.TraceType

Возвращает тип трассировки этого объекта.

Тип трассировки используется для построения сигнатуры tf.function при трассировке и для сопоставления аргументов с существующими сигнатурами. Когда вызывается объект Function, tf.function анализирует тип трассировки аргументов вызова. Если существует существующая сигнатура соответствующего типа, она будет использована. В противном случае будет отслеживаться новая функция, и её сигнатура будет использовать тип трассировки аргументов вызова.