如何使用tf.compute_gradient进行渐变?

时间:2016-04-20 15:26:47

标签: tensorflow

我正在尝试通过以下方式计算渐变的值:

grads_vars=optimizer.compute_gradients(loss,tf.trainable_variables())

函数'compute_gradients'返回一个两列列表,其中包含一些tf.gradient事物和一些tf.variable事物。当我尝试获取此列表时出现错误,因为它不是张量。

1 个答案:

答案 0 :(得分:0)

grads_and_vars是张量[(gradient1, variable1), ...]

的元组列表

但您只能获取[gradient1, gradient2, ...]等张量列表,因此您需要将第一个列表转换为第二种格式