如何理解TensorFlow 2.0中函数签名与图结构的优化建议?
- 内容介绍
- 文章标签
- 相关推荐
本文共计803个文字,预计阅读时间需要4分钟。
input_signature的妙处:1. 可以为函数限定输入类型,防止调用函数时输入错误;2. 函数有input_signature后,在tensorflow中才能保存成savedmodel。保存savedmodel过程中,需要使用保存成savedmodel的步骤:。
input_signature的好处:
1.可以限定函数的输入类型,以防止调用函数时调错,
2.一个函数有了input_signature之后,在tensorflow里边才可以保存成savedmodel。在保存成savedmodel的过程中,需要使用get_concrete_function函数把一个tf.function标注的普通的python函数变成带有图定义的函数。
下面的代码具体体现了input_signature可以限定函数的输入类型这一作用。
本文共计803个文字,预计阅读时间需要4分钟。
input_signature的妙处:1. 可以为函数限定输入类型,防止调用函数时输入错误;2. 函数有input_signature后,在tensorflow中才能保存成savedmodel。保存savedmodel过程中,需要使用保存成savedmodel的步骤:。
input_signature的好处:
1.可以限定函数的输入类型,以防止调用函数时调错,
2.一个函数有了input_signature之后,在tensorflow里边才可以保存成savedmodel。在保存成savedmodel的过程中,需要使用get_concrete_function函数把一个tf.function标注的普通的python函数变成带有图定义的函数。
下面的代码具体体现了input_signature可以限定函数的输入类型这一作用。

