提交 2d42e182 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

correct assertions

上级 c7b04616
...@@ -815,9 +815,9 @@ def hessian(cost, wrt, consider_constant=None, warn_type=False, ...@@ -815,9 +815,9 @@ def hessian(cost, wrt, consider_constant=None, warn_type=False,
hessians = [] hessians = []
for input in wrt: for input in wrt:
assert isinstance(cost, TensorVariable), \ assert isinstance(input, TensorVariable), \
"tensor.hessian expects a (list of) Tensor Variable as `wrt`" "tensor.hessian expects a (list of) Tensor Variable as `wrt`"
assert cost.ndim == 0, \ assert input.ndim == 0, \
"tensor.hessian expects a (list of) 1 dimensional variable"\ "tensor.hessian expects a (list of) 1 dimensional variable"\
"as `wrt`" "as `wrt`"
expr = grad(cost, input) expr = grad(cost, input)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论