我使用tf.Estimator
训练了一个简单的自动编码器。在训练期间,更新特定张量W
,其中W
是矩阵。培训结束后,我想检索W
并使用numpy
读取其值。
如果我没有使用tf.Estimator
,这是一项简单的任务,我会调用.eval()
并通过我的会话。但是,Estimator
是一个高级API,会话的初始化和使用都是在幕后完成的。
我还尝试使用Estimator.predict
和EstimatorSpec
返回W
,但似乎无效。我收到以下错误:
TypeError:预期单个Tensor时的张量列表。
在使用numpy
进行培训后,是否可以直接检索张量的tf.Estimator
值。如果是这样,怎么样?