如何使用 tflite 计算矩阵的逆矩阵

Question

据我所知，矩阵的逆是一个公算子。

当tf.raw_ops。MatrixInverse在tflite中不支持，BatchMatrixInverse在GraphDef版本1205中不可用。

我如何计算矩阵的逆在tflite?

祝福

Answer 1

在将模型转换为tflite模型之前，必须使用以下代码

使TF ops能够在tflite中工作

converter.target_spec.supported_ops = [
tf.lite.OpsSet.TFLITE_BUILTINS,  # enable TensorFlow Lite ops.
tf.lite.OpsSet.SELECT_TF_OPS  # enable TensorFlow ops.
]

详细信息请参阅本文档。谢谢你。