提交 42049ba4 authored 作者: Frederic's avatar Frederic

TrueDot, don't compare the grad_preserves_dense attribute

上级 e6258701
...@@ -2623,11 +2623,13 @@ class TrueDot(gof.op.Op): ...@@ -2623,11 +2623,13 @@ class TrueDot(gof.op.Op):
self.grad_preserves_dense = grad_preserves_dense self.grad_preserves_dense = grad_preserves_dense
def __eq__(self, other): def __eq__(self, other):
return (type(self) == type(other) and # The grad_preserves_dense attribute don't change the
self.grad_preserves_dense == other.grad_preserves_dense) # execution behavior. To have Theano merge optimizer merging
# them, we shouldn't compare it here.
return type(self) == type(other)
def __hash__(self): def __hash__(self):
return hash(type(self)) ^ hash(self.grad_preserves_dense) return hash(type(self))
def __ne__(self, other): def __ne__(self, other):
return not (self == other) return not (self == other)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论