tf.types.experimental.SupportsTracingProtocol

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

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

Методы

__tf_tracing_type__

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

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

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

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