简体   繁体   English

在张量流中初始化集合中的变量

[英]Initialize variables in collection in the tensorflow

how to initialize some variables in the collection? 如何初始化集合中的一些变量?

 with tf.Session() as sess:
    col1 = tf.add_to_collection('init',a )
    sess.run(tf.variables_initializer(col1))
    uninitializer_vars  = []
    for var in tf.global_variables():
        try :
            sess.run(var)
        except tf.errors.FailedPreconditionError:
            uninitializer_vars.append(var)
    print(uninitializer_vars)

Just pass the list of variables in a collection to https://www.tensorflow.org/api_docs/python/tf/variables_initializer to get an op that will initialize these variables. 只需将集合中的变量列表传递给https://www.tensorflow.org/api_docs/python/tf/variables_initializer即可获得将这些变量初始化的操作。

vars = tf.get_collection('my_collection_name')
vars_init_op = tf.variables_initializer(vars)
sess.run(vars_init_op)

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM