當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。