tensorflow中对指定变量进行初始化

最新推荐文章于 2022-07-27 10:51:24 发布

木里先森

最新推荐文章于 2022-07-27 10:51:24 发布

阅读量1.5k

点赞数

CC 4.0 BY-SA版权

分类专栏： python 深度学习

本文链接：https://blog.csdn.net/mr_muli/article/details/91313187

python 同时被 2 个专栏收录

176 篇文章

订阅专栏

深度学习

66 篇文章

订阅专栏

本文详细介绍了在TensorFlow中如何获取未初始化的变量集合，创建并初始化所有变量的方法。通过具体示例，展示了如何使用tf.report_uninitialized_variables()检查未初始化的变量，以及如何使用tf.variables_initializer()对特定变量进行初始化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

获取未初始化的变量集合
print(sess.run(tf.report_uninitialized_variables()))
例如我的：
[b’Conv2d_1b_1x1/weights’ b’Conv2d_1b_1x1/biases’ b’Conv2d_2a_5x5/weights’ b’Conv2d_2a_5x5/biases
创建所有变量集合：lac_var=[]
获取所有变量： [lac_var.append(i) for i in tf.global_variables()]
打印：print(lac_var)
对指定变量进行初始化：sess.run(tf.variables_initializer(lac_var[0:2]))

#  获取未初始化的变量集合
#  [b'Conv2d_1b_1x1/weights' b'Conv2d_1b_1x1/biases' b'Conv2d_2a_5x5/weights' b'Conv2d_2a_5x5/biases']
  print(sess.run(tf.report_uninitialized_variables()))
  lac_var=[]
  [lac_var.append(i) for i in tf.global_variables()]
  print(lac_var)
  sess.run(tf.variables_initializer(lac_var[0:]))