Python tf.compat.v1.autograph.to_graph用法及代码示例

将 Python 实体转换为 TensorFlow 图。

用法

tf.compat.v1.autograph.to_graph(
    entity, recursive=True, arg_values=None, arg_types=None,
    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 foo(x):
    if x > 0:
      y = x * x
    else:
      y = -x
    return y

  converted_foo = to_graph(foo)

  x = tf.constant(1)
  y = converted_foo(x)  # converted_foo is a TensorFlow Op-like.
  assert is_tensor(y)

支持的 Python 实体包括：

• functions
• classes
• 对象方法

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

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

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