用法
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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。