python - 如何初始化tensorflow函数中定义的变量？

Question

在 TensorFlow 中，我想在函数内部定义一个变量，进行一些处理并根据一些计算返回一个值。但是，我无法在函数内部初始化变量。这是代码的最小示例:

import tensorflow as tf

def foo():
    x = tf.Variable(tf.ones([1]))
    y = tf.ones([1])
    return x+y

if __name__ == '__main__':
    with tf.Session() as sess:
        init = tf.global_variables_initializer()
        sess.run(init)
        print(sess.run(foo()))

运行代码会产生以下错误:

tensorflow.python.framework.errors_impl.FailedPreconditionError: Attempting to use uninitialized value Variable

Answer 1

在初始化所有变量之前，函数 foo() 根本没有被调用。所以它无法初始化 foo() 中的变量。我们需要在运行 session 之前调用该函数。

import tensorflow as tf

def foo():
    x=tf.Variable(tf.zeros([1]))
    y=tf.ones([1])
    return x+y

with tf.Session() as sess:
    result=foo()
    init=tf.global_variables_initializer()
    sess.run(init)
    print(sess.run(result))