我刚读完了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.
意味着什么
答案 0 :(得分:1)
假设你的损失函数是亏损的 和x是你想要渐变的值。
假设你想要在chainrule中乘以的函数输出是dz吗
tf.gradients(loss,x, grad_ys=dz)
我希望这能解决问题