使用Tensorflow执行链规则

时间:2016-08-31 22:36:00

标签: python-3.x tensorflow gradients

我刚读完了Tensorflow文档,发现tf.gradients(tf.gradients(ys, xs, grad_ys=None, name='gradients', colocate_gradients_with_ops=False, gate_gradients=False, aggregation_method=None)可用于计算与dy/dx相关的渐变。目前,我想计算dz(f(x))/dx的渐变(应使用链规则将其分解为dz/df * df/dx),我想知道Tensorflow中是否有办法执行此链规则。我也想知道Tensorflow Doc。{/ p>中的grad_ys is a list of Tensor, holding the gradients received by the ys. The list must be the same length as ys.意味着什么

1 个答案:

答案 0 :(得分:1)

假设你的损失函数是亏损的 和x是你想要渐变的值。

假设你想要在chainrule中乘以的函数输出是dz吗

tf.gradients(loss,x, grad_ys=dz)

我希望这能解决问题