gpt4 book ai didi

python - tf.Estimator 训练后检索张量 (Numpy) 值

转载 作者:行者123 更新时间:2023-12-01 09:23:07 24 4
gpt4 key购买 nike

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

如果我没有使用tf.Estimator,这是一个简单的任务,我会调用.eval()并传递我的 session 。然而,Estimator 是一个高级 API, session 的初始化和使用都是在后台完成的。

我还尝试使用 Estimator.predictEstimatorSpec 返回 W 但它似乎不起作用。我收到以下错误:

TypeError: List of Tensors when single Tensor expected.

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

最佳答案

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

关于python - tf.Estimator 训练后检索张量 (Numpy) 值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50649236/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com