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

R operator of DimShuffle

上级 8476ebe8
...@@ -218,6 +218,13 @@ class DimShuffle(Op): ...@@ -218,6 +218,13 @@ class DimShuffle(Op):
rval.insert(augm, 1) rval.insert(augm, 1)
return [rval] return [rval]
def R_op(self, inputs, eval_points):
if None in eval_points:
return [None]
return self.make_node(*eval_points).outputs
def c_code(self, node, name, inp, out, sub): def c_code(self, node, name, inp, out, sub):
input, = inp input, = inp
res, = out res, = out
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论