tensorflow服务是否适用于多个输入(或输出)?

时间:2018-01-30 10:58:18

标签: python tensorflow keras tensorflow-serving

根据:https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html#exporting-a-model-with-tensorflow-serving

  

任何Keras模型都可以使用TensorFlow服务导出(只要它   只有一个输入和一个输出,这是一个限制   TF-服务

这(仍然)是真的吗?我找不到其他参考资料

1 个答案:

答案 0 :(得分:2)

只要您使用TF-serving的PREDICT接口,就可以拥有多个输入和输出张量,并在SavedModel的不同签名中公开它们的任意组合。例如,请查看

https://www.tensorflow.org/serving/signature_defs

因此虽然编写的Keras-exporter代码可能存在限制(我还没看过),但在TensorFlow服务的SavedModel中没有这样的限制(我正在处理的产品经常使用该功能)