Python tf.lite.experimental.authoring.compatible用法及代码示例

使用 TFLite 兼容性检查将 tf.function 包装到可调用函数中。

用法

tf.lite.experimental.authoring.compatible(
    target=None, converter_target_spec=None, **kwargs
)

例子：

@tf.lite.experimental.authoring.compatible
@tf.function(input_signature=[
    tf.TensorSpec(shape=[None], dtype=tf.float32)
])
def f(x):
    return tf.cosh(x)

result = f(tf.constant([0.0]))
# COMPATIBILITY WARNING:op 'tf.Cosh' require(s) "Select TF Ops" for model
# conversion for TensorFlow Lite.
# Op:tf.Cosh
#   - tensorflow/python/framework/op_def_library.py:748
#   - tensorflow/python/ops/gen_math_ops.py:2458
#   - <stdin>:6

警告：实验接口，如有更改。