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 參考文檔。