tf.autograph.to_graph

Преобразует сущность Python в граф TensorFlow.

tf.autograph.to_graph(
    entity, recursive=True, experimental_optional_features=None
)

См. также: tf.autograph.to_code, tf.function.

В отличие от tf.function, to_graph — низкоуровневый транспилятор, который преобразует код Python в код графа TensorFlow. Он не реализует кэширование и управление переменными и не создает фактические операции. Его лучше использовать, когда требуется больший контроль над создаваемым графом TensorFlow. Еще одно отличие от tf.function состоит в том, что to_graph не оборачивает граф в функцию TensorFlow или вызываемый объект Python. Внутри tf.function использует to_graph.

Пример использования:

def f(x):
  if x > 0:
    y = x * x
  else:
    y = -x
  return y

converted_f = to_graph(f)
x = tf.constant(2)
converted_f(x)  # converted_foo is like a TensorFlow Op.
<tf.Tensor: shape=(), dtype=int32, numpy=4>

Поддерживаются следующие сущности Python:

• функции
• классы
• методы объектов

Функции преобразуются в новые функции с преобразованным кодом.

Классы преобразуются путем создания нового класса, методы которого используют преобразованный код.

Методы преобразуются в несвязанные функции с дополнительным первым аргументом под названием self.

См. руководство по tf.function и AutoGraph. Более подробную информацию см. в справочной документации AutoGraph.