提交 03a22e2c authored 作者: Kyung Hyun Cho's avatar Kyung Hyun Cho

moved dtype_eps outside

上级 15522a96
...@@ -78,13 +78,12 @@ class multiple_outputs_numeric_grad: ...@@ -78,13 +78,12 @@ class multiple_outputs_numeric_grad:
if not ndarray_mask: if not ndarray_mask:
ndarray_mask = [True for x in pt] ndarray_mask = [True for x in pt]
dtype_eps = multiple_outputs_numeric_grad.type_eps['float64'] dtype_eps = type_eps['float64']
for i, p in enumerate(pt): for i, p in enumerate(pt):
if ndarray_mask[i]: if ndarray_mask[i]:
pt[i] = numpy.array(p) pt[i] = numpy.array(p)
_eps = multiple_outputs_numeric_grad.type_eps[str( _eps = type_eps[str(pt[i].dtype)]
pt[i].dtype)]
if _eps > dtype_eps: if _eps > dtype_eps:
dtype_eps = _eps dtype_eps = _eps
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论