提交 a7d36d53 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Merge

...@@ -1389,6 +1389,10 @@ class Split(Op): ...@@ -1389,6 +1389,10 @@ class Split(Op):
def __init__(self, len_splits): def __init__(self, len_splits):
self.len_splits = int(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): def make_node(self, x, axis, splits):
"""WRITEME""" """WRITEME"""
x = as_tensor(x) x = as_tensor(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论