提交 5e6005f8 authored 作者: Frederic's avatar Frederic

Fix equal_computations with NoneType, as it don't have dtype/shape attribute.

上级 7ab32ffb
...@@ -448,9 +448,7 @@ def equal_computations(xs, ys, in_xs=None, in_ys=None): ...@@ -448,9 +448,7 @@ def equal_computations(xs, ys, in_xs=None, in_ys=None):
if dx != dy: if dx != dy:
if (isinstance(dx, tensor.Constant) and if (isinstance(dx, tensor.Constant) and
isinstance(dy, tensor.Constant)): isinstance(dy, tensor.Constant)):
if not (numpy.all(dx.data == dy.data) and if not dx.equals(dy):
dx.type.dtype == dy.type.dtype and
dx.data.shape == dy.data.shape):
return False return False
else: else:
pass pass
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论