Tensorflow定义了一个集合的预设，可以在这里看到:https://www.tensorflow.org/versions/r0.12/api_docs/python/framework/graph_collections
我目前正在使用tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES)来获取所有变量[*命名为；如果它们不存在，那么即使存在，也不会显示它们]。
同样，我希望tf.get_collection(tf.GraphKeys.WEIGHTS)输出一个权重列表，但是它是一个空数组。对于GraphKeys.BIASES和.ACTIVATIONS也是如此。
这里发生了什么？
我认为这里似乎有两种可能性。首先，这些实际上是不会自动定义的，而只是推荐的集合名称。其次，我的网络非常困惑，但事实并非如此。
有人对此有经验吗？
最佳答案
默认情况下，所有变量都绑定(bind)到tf.GraphKeys.GLOBAL_VARIABLES集合。方便的方法是将每个权重设置为集合tf.GraphKeys.WEIGHTS，如下所示:
In [2]: w = tf.Variable([1,2,3], collections=[tf.GraphKeys.WEIGHTS], dtype=tf.float32)
In [3]: w2 = tf.Variable([11,22,32], collections=[tf.GraphKeys.WEIGHTS], dtype=tf.float32)
然后，您可以通过以下方式获取它们:
tf.get_collection_ref(tf.GraphKeys.WEIGHTS)
这是权重:
[<tf.Variable 'Variable:0' shape=(3,) dtype=float32_ref>,
<tf.Variable 'Variable_1:0' shape=(3,) dtype=float32_ref>]
关于python - 如何使用TensorFlow Graphkeys获得所有权重？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/45256627/