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

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

用法

tf.compat.v1.autograph.to_code(
    entity, recursive=True, arg_values=None, arg_types=None, indentation='
    ', experimental_optional_features=None
)

参数

entity Python 可调用或类。
recursive 是否递归转换转换后的函数可能调用的任何函数。
arg_values 已弃用。
arg_types 已弃用。
indentation 已弃用。
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.autograph.to_graph。

注意：如果函数已用 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.compat.v1.autograph.to_code。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。