在tf.Estimator训练后检索Tensor(Numpy)值

时间:2018-06-01 18:32:33

标签: python numpy tensorflow tensorflow-estimator

我使用tf.Estimator训练了一个简单的自动编码器。在训练期间,更新特定张量W,其中W是矩阵。培训结束后,我想检索W并使用numpy读取其值。

如果我没有使用tf.Estimator,这是一项简单的任务,我会调用.eval()并通过我的会话。但是,Estimator是一个高级API,会话的初始化和使用都是在幕后完成的。

我还尝试使用Estimator.predictEstimatorSpec返回W,但似乎无效。我收到以下错误:

  

TypeError:预期单个Tensor时的张量列表。

在使用numpy进行培训后,是否可以直接检索张量的tf.Estimator值。如果是这样,怎么样?

1 个答案:

答案 0 :(得分:1)

假设W作为变量存储在模型中,您可以使用get_variable_value对象的Estimator方法。请参阅here