tensorflow - tf.initialize_all_variables() 和 tf.initialize_local_variables() 有什么区别？

Question

我正在查看此示例中的代码:fully_connected_reader.py

我对第 147 行和第 148 行感到困惑:

init_op = tf.group(tf.initialize_all_variables(),
                   tf.initialize_local_variables())

我不知道哪些变量是所有变量，哪些是局部变量。有什么想法吗？

Answer 1

tf.initialize_all_variables() 是 tf.initialize_variables(tf.all_variables()) 的快捷方式，tf.initialize_local_variables() 是tf.initialize_variables(tf.local_variables()) 的快捷方式，分别初始化 GraphKeys.VARIABLES 和 GraphKeys.LOCAL_VARIABLE 集合中的变量。

GraphKeys.LOCAL_VARIABLES 集合中的变量是添加到图表中但未保存或恢复的变量 ( source )。

tf.Variable()默认情况下会向 GraphKeys.VARIABLE 集合添加一个新变量，该变量可以通过 collections= argument 进行控制。