当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.lite.Interpreter.get_signature_runner用法及代码示例


用法

get_signature_runner(
    signature_key=None
)

参数

  • signature_key SignatureDef 的签名 key ,当且仅当模型具有单个 SignatureDef 时,它才能为 None。默认值为无。

返回

  • 这将返回一个可调用的,可以运行由参数 'signature_key' 定义的 SignatureDef 的推理。可调用对象将采用与 SignatureDef 的参数相对应的关键参数,该参数应该具有 numpy 值。可调用将返回从输出名称映射到计算结果的 numpy 值的字典。

抛出

  • ValueError 如果通过 signature_key 无效。

获取可调用以推断特定 SignatureDef。

示例用法,

interpreter = tf.lite.Interpreter(model_content=tflite_model)
interpreter.allocate_tensors()
fn = interpreter.get_signature_runner('div_with_remainder')
output = fn(x=np.array([3]), y=np.array([2]))
print(output)
# {
#   'quotient':array([1.], dtype=float32)
#   'remainder':array([1.], dtype=float32)
# }

如果模型只有一个签名,则不能为 signature_key 传递任何内容。

所有使用的名称都是这个特定的 SignatureDef 名称。

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.lite.Interpreter.get_signature_runner。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。