Tensorflow 将经过训练的keras模型的批量大小设置为1

我在自己的数据集上训练了一个keras模型。但是，在加载重量后，摘要将无显示为第一个维度（批次大小）

---

我想知道将形状固定为批量大小1的过程，因为我必须将其固定，以便我可以使用GPU支持将模型转换为tflite。

而不是使用

input\u shape

来输入模型和训练，使用

batch\u input\u shape

并选择所需的

batch\u size

。有关更多信息，请参阅。谢谢