如何从tf.report_uninitialized_variables获取未初始化的变量列表

时间:2017-05-30 17:14:53

标签: tensorflow

文档说它是1 d张量,但是,我没有弄清楚如何访问列表。

我更喜欢实际的变量而不是名字,因为我想通过tf.variables_initializer()初始化它们

1 个答案:

答案 0 :(得分:7)

tf.report_uninitialized_variables()为您提供了一个包含变量名称的张量。所以它比我的solution here更丑陋(在我看来)。

您需要找到与report_uninitialized_variables所得名称相对应的所有变量,并在tf.variables_initializer()中使用它们。像这样:

tf.variables_initializer(
    [v for v in tf.global_variables() if v.name.split(':')[0] in set(sess.run(tf.report_uninitialized_variables()))
])