提交 5e26b8f3 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

more comprehensive rval type check for hessian

上级 d910c036
......@@ -835,6 +835,9 @@ def hessian(cost, wrt, consider_constant=None, warn_type=False,
sequences=arange(expr.shape[0]),
non_sequences=[expr, input])
hessians.append(hess)
if not use_list:
hessians = hessians[0]
return hessians
if use_list and not isinstance(hessians, (list, tuple)):
return [hessians]
elif not use_list and isinstance(hessians, (list, tuple)):
return hessianss[0]
else:
return hessians
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论