1. 首页
  2. 数据库
  3. 其它
  4. tensorflow2.0的函数签名与图结构

tensorflow2.0的函数签名与图结构

上传者: 2020-12-23 03:19:49上传 PDF文件 39.43KB 热度 9次
input_signature的好处: 1.可以限定函数的输入类型,以防止调用函数时调错, 2.一个函数有了input_signature之后,在tensorflow里边才可以保存成savedmodel。在保存成savedmodel的过程中,需要使用get_concrete_function函数把一个tf.function标注的普通的python函数变成带有图定义的函数。 下面的代码具体体现了input_signature可以限定函数的输入类型这一作用。 @tf.function(input_signature=[tf.TensorSpec([None], tf.int32, name='x
用户评论