如何使用tf.gradients和gradients_apply

时间:2017-09-15 12:45:18

标签: tensorflow tensorflow-serving

我使用(tf.vars是可训练的张量列表)来计算渐变

x = tf.gradients(y,tf.vars)

现在假设我想使用optimizer.apply_gradients来应用这个渐变,然后我需要给出一个包含(渐变,变量)形式的元组的列表。 我如何生成这样的列表?

optimizer.apply_gradients(??)

空间里有什么?

1 个答案:

答案 0 :(得分:0)

要传递给optimizer.apply_gradients的列表应该是(g, v)对的列表,其中v是图的变量,g是渐变w.r.t.您想要更新v

所以这样的事情应该有效:

[(g, v) for g, v in zip(x, tf.vars)]