python - tf.function 是否需要在内部函数中使用？

Question

在编写函数时，是否tf.function需要在所有内部函数中都使用装饰器才能获得好处？

@tf.function  # Necessary or not?
def add(a, b):
    return a + b

@tf.function
def compute(a, b):
    return add(a, b)

我在官方文档中找不到信息。他们确实提到使用内部装饰功能是可能的。

在不了解内部结构的情况下，我希望任何仅包含 TF 操作的函数调用都将合并到由最外层函数创建的图形中。拥有像不被装饰这样的内部函数add将是一个很大的优势，因为它们也接受例如 numpy 类型。

那么装修有必要吗？

标签： pythontensorflow