python - Tensorflow:tf.get_collection不返回范围中的变量

Question

我正在尝试获取变量范围内的所有变量，如here所述。但是，即使该范围中存在变量，tf.get_collection(tf.GraphKeys.VARIABLES, scope='my_scope')行也将返回一个空列表。

这是一些示例代码:

import tensorflow as tf

with tf.variable_scope('my_scope'):
    a = tf.Variable(0)
print tf.get_collection(tf.GraphKeys.VARIABLES, scope='my_scope')

打印 []。

如何获得在 'my_scope'中声明的变量？

Answer 1

自TensorFlow 0.12起，不推荐使用tf.GraphKeys.VARIABLES集合名称。使用tf.GraphKeys.GLOBAL_VARIABLES将产生预期的结果:

with tf.variable_scope('my_scope'):
    a = tf.Variable(0)
print tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='my_scope')
# ==> '[<tensorflow.python.ops.variables.Variable object at 0x7f33f67ebbd0>]'