如何对张量流中的所有全局变量和局部变量进行常量初始化?

时间:2018-09-03 17:45:48

标签: tensorflow deep-learning

我尝试通过

tf.variables_initializer(tf.constant_initializer(tf.global_variables()))

但是没有用。

有人可以帮我吗?

1 个答案:

答案 0 :(得分:0)

根据文档,tf.variables_initializer接受变量列表,而不是初始化程序。因此,如果要使用常量初始化变量,则可能需要执行以下步骤:

1)在构造变量时使用tf.constant_initializer,例如v = tf.get_variable(name='v', initializer=tf.constant_initializer(0.), shape=(1))

2)获取要初始化的变量列表,

3)设置init_op,即init_op = tf.variables_initializer([v1, v2, ...])

4)使用会话运行init_op。

我并不是说这是唯一的方法,但它应该是一种常见的方法。