如何在Tensorflow中更改已保存的模型输入形状？

import tensorflow as tf

# 加载已保存的模型
loaded_model = tf.saved_model.load('path/to/saved_model')

# 获取模型的输入签名
input_signature = loaded_model.signatures['serving_default'].inputs

# 获取输入张量的名称和形状
input_tensor_name = input_signature['input_tensor'].name
input_tensor_shape = input_signature['input_tensor'].shape

# 更改输入形状
new_input_shape = (None, 224, 224, 3)  # 新的输入形状
reshaped_input = tf.reshape(input_tensor_name, new_input_shape)

# 创建新的输入签名
new_input_signature = tf.TensorSpec(shape=new_input_shape, dtype=tf.float32, name='input_tensor')

# 更新模型的输入签名
loaded_model.signatures['serving_default'] = tf.saved_model.signature_def_utils.build_signature_def(
    inputs={'input_tensor': new_input_signature},
    outputs=loaded_model.signatures['serving_default'].outputs,
    method_name=loaded_model.signatures['serving_default'].method_name
)

# 保存更新后的模型
tf.saved_model.save(loaded_model, 'path/to/updated_model')