提交 dce31798 authored 作者: Frederic Bastien's avatar Frederic Bastien

bugfix the new option.

上级 befb9054
......@@ -2555,7 +2555,7 @@ def verify_grad(op, pt, n_tests=2, rng=None, eps=None, tol=None, mode=None, cast
if cast_to_output_type:
g_cost = cast(g_cost, o_output.dtype)
symbolic_grad = grad(cost, tensor_pt,as_tensor_variable(1.0,name='g_cost'))
symbolic_grad = grad(cost, tensor_pt, g_cost)
grad_fn = function(tensor_pt, symbolic_grad)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论