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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。