TensorFlow simple_save()为什么输入/输出字典是必需的?

时间:2018-12-17 21:06:10

标签: tensorflow tensorflow-serving

既然所有var / graph都已加载,为什么我需要向tf.saved_model.simple_save()提供输入和输出?

我尝试使用get_tensor_by_name()加载变量,该变量在输入/输出字典中未指定,并且可以正常工作!那么,为什么不让我有空白/无输入/输出并按变量名抓取我的变量呢?

1 个答案:

答案 0 :(得分:0)

指定模型的输入和输出张量时,将完全指定推理图。假设有一个输入只有两个输出的模型。例如,该模型可以预测明天的温度以及是否会下雨。也许我想为只给我温度的模型保存一个推理图。

当您指定ins和outs时,TensorFlow知道哪些层将它们连接起来。 get_tensor_by_name()适用于您的情况的原因可能是因为您获取了一个将输入连接到输出的层。