Python tf.autograph.to_code用法及代碼示例

以字符串形式返回 AutoGraph 生成的源代碼。

用法

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

entity Python 可調用或要轉換的類。
recursive 是否遞歸轉換轉換後的函數可能調用的任何函數。
experimental_optional_features None ，一個元組或單個 tf.autograph.experimental.Feature 值。

def f(x):
  if x < 0:
    x = -x
  return x
tf.autograph.to_code(f)
"...def tf__f(x):..."

注意：如果函數已用 tf.function 修飾，則傳遞其底層 Python 函數，而不是 `tf.function 創建的可調用函數：

@tf.function
def f(x):
  if x < 0:
    x = -x
  return x
tf.autograph.to_code(f.python_function)
"...def tf__f(x):..."

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.autograph.to_code。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。