Python 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 实体包括：

• functions
• classes
• 对象方法

使用转换后的代码将函数转换为新函数。

通过生成其方法使用转换代码的新类来转换类。

方法被转换为具有额外的第一个参数 self 的未绑定函数。

有关教程，请参阅 tf.function 和 AutoGraph 指南。有关更多详细信息，请参阅 AutoGraph 参考文档。