将keras模型导出为tf savedModel格式:如何修复serving_input_fn

import tensorflow as tf
from tensorflow import keras

model = keras.models.load_model('path_to_keras_model.h5')

tf_model = tf.keras.Sequential()

for layer in model.layers:
    tf_model.add(layer)

def serving_input_fn():
    inputs = {'input_name': tf.placeholder(tf.float32, [None, input_shape])}  # 替换input_name为实际的输入名称
    return tf.estimator.export.ServingInputReceiver(inputs, inputs)

export_path = 'path_to_export_model'  # 替换为实际的导出路径
tf.saved_model.simple_save(tf.keras.backend.get_session(), export_path, inputs={'input_name': model.input}, outputs={t.name: t for t in model.outputs}, serving_input_receiver_fn=serving_input_fn)