如何获取一个numpy数组作为在估计器上构建的Tensorflow程序中服务的输入?

时间:2018-10-29 21:25:46

标签: python numpy tensorflow

我有一个由Tensorflow估算器构建的CNN网络程序。在训练和评估部分,我有tf.estimator.inputs.numpy_input_fn()将numpy数组转换为张量。然后,在模型函数中,我将使用tf.reshape从tf.estimator.inputs.numpy_input_fn()重塑结果张量。但是,我想保存该模型以进行投放。当我需要构建serving_input_receiver_function时,我不知道该怎么办。由于服务的输入也将是numpy数组,但是模型函数将使用张量。似乎需要像在训练和评估部分中那样将numpy数组转换为张量。但是我不知道如何在serve_input_receiver_function中做到这一点。 (我在程序中根本没有使用tf.placeholder。)

0 个答案:

没有答案