c - 如何在 Tensorflow Lite(实验性 C API)中创建输入张量并与解释器一起使用？

Question

我们如何使用 C API 在 Tensorflow Lite 中创建张量？然后我们如何将它们与解释器一起使用？

我知道 Tensorflow Lite 有 C++ API 的文档，而常规的 Tensorflow 有 C API 的文档。但是，几乎没有关于使用 C API for Lite 的信息。

Lite C++ API 有一个创建张量的函数，但我在 Lite 的 C API 头文件(或任何其他相关头文件/源文件)中没有看到它。

以下存在于使用常规 Tensorflow C API 的示例中。但它似乎并不存在于 TF Lite C API 中。

TF_Tensor* input_tensor = tf_utils::CreateTensor(TF_FLOAT, input_dims.data(), input_dims.size(), input_vals.data(), input_vals.size() * sizeof(float));

Answer 1

TensorFlow Lite 张量 (TFL_Tensor) 由解释器 (TFL_Interpreter) 实例拥有，不能由用户创建。您可以使用以下方法获取输入/输出张量的句柄:

 TFL_Tensor* TFL_InterpreterGetInputTensor(
    const TFL_Interpreter* interpreter, int32_t input_index);

 const TFL_Tensor* TFL_InterpreterGetOutputTensor(
    const TFL_Interpreter* interpreter, int32_t output_index);

可以使用 c_api.h 中描述的方法读取(或更新)张量数据 header 。缺少文档，因为这仍然是一个实验性 API，但我们希望在不久的将来将其移出实验性。