提交 d1277aa2 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

R operator for split

上级 ca0250cf
...@@ -3788,6 +3788,10 @@ class Split(Op): ...@@ -3788,6 +3788,10 @@ class Split(Op):
_, axis, _ = inputs _, axis, _ = inputs
return [join(axis, *g_outputs), None, None] return [join(axis, *g_outputs), None, None]
def R_op(self, inputs, eval_points):
if eval_points[0] is None:
return [None for i in self.len_splits]
return self.make_node(eval_points[0], *inputs[1:]).outputs
class Rebroadcast(Op): class Rebroadcast(Op):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论