python - 重塑操作引发错误“num_input_elements！= num_output_elements”（50！= 1）

Question

当我将自己的模型替换为 hello world 模型时，当输入为 2D 张量且输出为 1D 张量时，reshape ops 会引发错误“num_input_elements != num_output_elements” (50 != 1)。

reshape input: int32[1,50]
reshape output: int32[1]
reshape shape value = -1

但是当我追踪代码时：tensorflow/lite/micro/kernels/reshape.cc: 56 int value = output_shape->data[i];

该值始终为 1

那么我该如何解决这个问题？谢谢。

标签： pythontensorflowtensorflow-lite