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 的未綁定函數。