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

R operator for rebroadcast

上级 d1277aa2
...@@ -3852,6 +3852,13 @@ class Rebroadcast(Op): ...@@ -3852,6 +3852,13 @@ class Rebroadcast(Op):
return [tuple(l)] return [tuple(l)]
def R_op(self, inputs, eval_points):
if eval_points[0] is None:
return [None]
return self.make_node(*eval_points).outputs
def addbroadcast(x, *axes): def addbroadcast(x, *axes):
""" """
Make the input broadcastable in the specified axes. Make the input broadcastable in the specified axes.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论