tf.function如何编译一个带有签名的python函数？

Question

如何将对张量操作的python函数编译成图形，尤其是wrt签名？文档不详细介绍tf.function

tf.function构造一个可调用函数，该可调用函数执行 TensorFlow 图 （），该图是通过对 中的 TensorFlow 操作进行跟踪编译而创建的，有效地执行为 TensorFlow 图。tf.Graphfuncfunc

它是否使用条件 （） 和循环 （） 调用的特殊方法来“跟踪”函数的实现？例如__bool____iter__

import tensorflow as tf

@tf.function

def op(t: tf.Tensor) -> tf.Tensor:

    if tf.reduce_sum(t) == 0:

        for _ in t:

            ...

可以使用结果和结果ifTensor.__bool__(...)for _ in tTensor.__iter__(...)

Answer 1

嗯 - 它比这更复杂。 本身就是一个完整的“子包”，所以当你调用 时，引擎盖下会发生很多事情。一个很好的起点可能在这里：https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/autograph/g3doc/reference。如果您真的好奇，您也可以直接深入研究代码。[e-mail]