tensorflow - 使用 Tensorflow 训练时修改张量的值

Question

我想在使用 Tensorflow 训练模型时修改张量的值。

这个张量是我模型中的张量之一

weight = tf.Variable(np_matrix)

经过一些迭代， weight 的值会自动更新。

我的问题是:如何修改 weight 的值非自动。我试过这种方法，但没有奏效。

modify_weight = sess.run([weight], feed_dict = feed_dict)
modify_weight[0] = [0, 0]
weight = tf.Variable(modify_weight)

这部分代码在 tf.Session()部分(因为我想在训练期间修改该值。)

谢谢!

Answer 1

和其他一切一样，赋值也是一个操作，我们必须用 tf.assign 创建一个图。并在 session 中运行它。

所以你创建一个这样的操作:

assign = tf.assign(weight, value)

哪里 value是 numpy具有相同形状的数组 weight (或可以使用提要字典修改的 tf.Placeholder)然后在 session 中运行此图:

sess.run(assign)

tf.Variable还有一个方法 assign ，因此您可以直接从变量开始创建操作:

assign = weight.assign(value)

然后在 session 中运行它。