将TensorFlow中的变量重新初始化为自定义值

Question

我想使用自定义w_gate初始化np.array张量，如下面的代码所示：

    w_init = np.ones(shape=(dim, self.config.nmodels)) / self.config.nmodels

    w_gate = tf.Variable(
        name="W",
        initial_value=w_init,
        dtype=tf.float32)

每隔一定数量的列车迭代，我希望w_gate再次重新初始化为w_init数组。为此，我根据Re-initialize variables in Tensorflow尝试了

sess.run(tf.variables_initializer([w_gate]))

在我的训练循环中。每一定次数的迭代执行该行。虽然w_gate似乎没有重新初始化。我在这里缺少什么？

Answer 1

你可以试试看吗？

w_gate_assign = tf.assign(w_gate, w_init)
sess.run(w_gate_assign)