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

Question

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

    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数组。 为此，基于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)