从TF转换为TFlite期间张量失去尺寸

时间:2019-08-19 11:15:13

标签: tensorflow tensorflow-lite

我正在将张量流模型转换为tflite模型(通过python脚本),尽管所有输入张量都作为tflite中的输入传递,但tflite中的某些张量没有任何尺寸。

因此在TF中,我有5个输入张量(4D,4D,1D,1D,1D),当我将其转换为tflite时,我得到了5个张量(4D,4D,1D,0D,0D)。输出张量没有问题

我认为toco_tooling.cc或其他文件中发生了某些事情

有帮助吗?

0 个答案:

没有答案