Python 在tf.估计器训练后检索张量(Numpy)值
我用Python 在tf.估计器训练后检索张量(Numpy)值,python,numpy,tensorflow,tensorflow-estimator,Python,Numpy,Tensorflow,Tensorflow Estimator,我用tf.Estimator训练了一个简单的自动编码器。在训练期间,一个特定的张量W被更新,其中W是一个矩阵。培训后,我想检索W,并使用numpy读取其值 这是一个简单的任务,如果我没有使用tf.Estimator,我会调用.eval()并通过我的会话。然而,Estimator是一个高级API,会话的初始化和使用都是在幕后完成的 我还尝试使用Estimator.predict和Estimator spec返回W,但它似乎不起作用。我得到以下错误: TypeError:需要单个张量时的张量列表 在
tf.Estimator
训练了一个简单的自动编码器。在训练期间,一个特定的张量W
被更新,其中W
是一个矩阵。培训后,我想检索W
,并使用numpy
读取其值
这是一个简单的任务,如果我没有使用tf.Estimator
,我会调用.eval()
并通过我的会话。然而,Estimator
是一个高级API,会话的初始化和使用都是在幕后完成的
我还尝试使用Estimator.predict
和Estimator spec
返回W
,但它似乎不起作用。我得到以下错误:
TypeError:需要单个张量时的张量列表
在使用
tf.Estimator
进行训练后,是否可以直接检索张量的numpy
值。如果是,如何进行?假设W
作为变量存储在模型中,您可以使用估计器
对象的获取变量值
方法。请参阅。嗨,欢迎来到StackOverflow!提供一个可能的代码示例将使某人更有可能帮助您。如果可能的话,提供一份报告。