用法
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 名稱。
相關用法
- Python tf.lite.Interpreter.get_signature_list用法及代碼示例
- Python tf.lite.Interpreter.tensor用法及代碼示例
- Python tf.lite.Interpreter.resize_tensor_input用法及代碼示例
- Python tf.lite.Interpreter用法及代碼示例
- Python tf.lite.experimental.QuantizationDebugger用法及代碼示例
- Python tf.lite.TFLiteConverter用法及代碼示例
- Python tf.lite.experimental.authoring.compatible用法及代碼示例
- Python tf.lite.experimental.load_delegate用法及代碼示例
- Python tf.linalg.LinearOperatorFullMatrix.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorToeplitz.solve用法及代碼示例
- Python tf.linalg.LinearOperatorIdentity.solvevec用法及代碼示例
- Python tf.linalg.LinearOperatorPermutation.solve用法及代碼示例
- Python tf.linalg.band_part用法及代碼示例
- Python tf.linalg.LinearOperatorKronecker.diag_part用法及代碼示例
- Python tf.linalg.lu_matrix_inverse用法及代碼示例
- Python tf.linalg.LinearOperatorToeplitz.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorBlockLowerTriangular.solvevec用法及代碼示例
- Python tf.linalg.LinearOperatorLowerTriangular.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorCirculant2D.solve用法及代碼示例
- Python tf.linalg.LinearOperatorCirculant3D.diag_part用法及代碼示例
注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.lite.Interpreter.get_signature_runner。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。