函数签名由函数原型组成。它告诉你的是关于函数的一般信息,它的名称,参数,它的范围以及其他杂项信息。
我们使用input_signature
对tf.function
修饰的函数进行数字签名;
tf.TensorSpec()
:#TensorSpec
: 描述一个张量。
tf.TensorSpec ( shape, dtype=tf.dtypes.float32, name=None )
对于被tf.function
修饰过的函数都有get_concrete_function
的属性,
可以通过该操作对函数添加函数签名,从而获取特定追踪。通过增加函数签名之后才能够将模型保存。