提交 1d54538d authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Implemented the __eq__ method for Split Op

上级 b2830444
......@@ -1388,7 +1388,11 @@ class Split(Op):
def __init__(self, len_splits):
self.len_splits = int(len_splits)
def __eq__(self, other):
return (type(self) == type(other) and
self.len_splits == other.len_splits)
def make_node(self, x, axis, splits):
"""WRITEME"""
x = as_tensor(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论