TF -如何正确设置用于Docker服务的模型签名？

import tensorflow as tf
from tensorflow.python.saved_model import signature_constants
from tensorflow.python.saved_model import tag_constants

model = tf.saved_model.load('/path/to/model')

input_tensor = model.signatures[signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY].inputs
output_tensor = model.signatures[signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY].outputs

docker_model = tf.saved_model.builder.SavedModelBuilder('/path/to/docker/model')
docker_model.add_meta_graph_and_variables(
    sess=tf.compat.v1.Session(),
    tags=[tag_constants.SERVING],
    signature_def_map={
        signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: tf.saved_model.signature_def_utils.predict_signature_def(
            inputs=input_tensor,
            outputs=output_tensor
        )
    }
)
docker_model.save()