python - 给定现有的 Keras 模型，有没有办法修复输入批量大小？

Question

我有一个以 hdf5 格式 (.h5) 保存的 Keras 模型 (TensorFlow 2.0)。我想加载这个现有模型并更改其输入形状，因此它只接受固定的批量大小。

输入批量大小可以通过 layer 来固定tensorflow.keras.Input(shape=..., batch_size=...)。但是，我不想从头开始创建新模型（我想使用已经存在的模型）。有没有办法使用 Python TF2 api 做到这一点，而无需从头开始创建模型并且不添加新层？

标签： pythonkerastensorflow2.0