tensorflow - 张量在从 TF 到 TFlite 的转换过程中丢失了它们的维度

Question

我正在将张量流模型转换为 tflite 模型（通过 python 脚本），尽管所有输入张量都作为输入在 tflite 中传递，其中一些在 tflite 中没有任何维度。

因此，在 TF 中，我有 5 个输入张量（4D、4D、1D、1D、1D），当我将其转换为 tflite 时，我得到 5 个张量（4D、4D、1D、0D、0D）。我对输出张量没有问题

我在想内部toco_tooling.cc或其他文件中发生了一些事情

一些帮助？

标签： tensorflowtensorflow-lite