我使用(tf.vars是可训练的张量列表)来计算渐变
x = tf.gradients(y,tf.vars)
现在假设我想使用optimizer.apply_gradients来应用这个渐变,然后我需要给出一个包含(渐变,变量)形式的元组的列表。 我如何生成这样的列表?
optimizer.apply_gradients(??)
空间里有什么?
答案 0 :(得分:0)
要传递给optimizer.apply_gradients的列表应该是(g, v)
对的列表,其中v
是图的变量,g
是渐变w.r.t.您想要更新v
。
所以这样的事情应该有效:
[(g, v) for g, v in zip(x, tf.vars)]