1、结构为字典类复核类型时的表示方式:
@tf.function(input_signature=({"key1":tf.TensorSpec(shape=(None,),dtype=tf.dtypes.string,name="key1","key2":tf.TensorSpec(shape=(None,),dtype=tf.dtypes.string,name="key2"))},))
表示入参结构如下:
{
"key1":["a","b"],
"key2":["e","f"]
}
注意事项:1、key1和key2代表的属性字段名必须小写
2、key1和key2对应的维度必须一致
2、输入的结构如果为简单的数组性,或多层数组:
@tf.function(input_signature=(tf.TensorSpec(shape=(None,None), dtype=tf.dtypes.string,name="key1"),))
表示的入参结构如下:
{
"key1":[['a','b'],['c','d']]
}
3、输入参数为多个参数时,
@tf.function(input_signature=(tf.TensorSpec(shape=(None,None), dtype=tf.dtypes.string,name="key1"),tf.TensorSpec(shape=(None,None), dtype=tf.dtypes.string,name="key2")))
{
"key1":[["a","b"]],
"key2":[["e","f"]]
}
本文详细介绍了TensorFlow中@tf.function输入签名的使用方法,包括字典类复核类型、简单数组和多层数组的表示方式。强调了参数结构的注意事项,如属性字段名的小写规则和维度一致性。此外,还展示了多个参数输入的情况。
797

被折叠的 条评论
为什么被折叠?



