提交 21620c33 authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

pass consider_constant to grad

上级 d047b81a
...@@ -1666,9 +1666,11 @@ def jacobian(expression, wrt, consider_constant=None, ...@@ -1666,9 +1666,11 @@ def jacobian(expression, wrt, consider_constant=None,
if expression.ndim == 0: if expression.ndim == 0:
# expression is just a scalar, use grad # expression is just a scalar, use grad
return format_as( return format_as(using_list, using_tuple,
using_list, using_tuple, grad(expression,
grad(expression, wrt, disconnected_inputs=disconnected_inputs)) wrt,
consider_constant=consider_constant,
disconnected_inputs=disconnected_inputs))
def inner_function(*args): def inner_function(*args):
idx = args[0] idx = args[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论