In Tensorflow, können Sie Ihren eigenen Sammlungsnamen definieren?

Question

Ich suchte alle Ressourcen in Tensorflow-API-Dokumente und kann keine Hinweise finden.

x=tf.get_variable('x',[2,2],collections='my_scope') 

aber bekommen nur leere Liste, wenn dabei:

tf.get_collection('my_scope') 

Answer 1

Sammlung S eine list der braucht Es scheint, als get_variable() verwendet wird, kann ich einen bestimmten Namen für Sammlungen Begriff wie setzen Sammlungsname.

>>x = tf.get_variable('x',[2,2], collections=['my_scope']) 
>>tf.get_collection('my_scope') 

[<tensorflow.python.ops.variables.Variable at 0x10d8e1590>] 

Achten Sie darauf, dass einige andere Operationen Nebenwirkungen haben können, wenn Sie es verwenden. wie tf.all_variables() wird nicht funktionieren und somit tf.initialize_all_variables() auch nicht Ihre Variable sehen. Eine Möglichkeit, dies zu beheben, besteht darin, auch die Standardsammlung anzugeben.

>>x = tf.get_variable('x',[2,2], collections=['my_scope', tf.GraphKeys.VARIABLES]) 

aber die Dinge beginnen zu langweilig.

Answer 2

Eigentlich kann man tf.get_collection verwenden, um eine neue Sammlung zu erstellen:

tf.get_collection('my_collection') 
var = tf.get_variable('var', [2, 2], initializer=tf.constant_initializer()) 
tf.add_to_collection('my_collection', var)