提交 95f550fd authored 作者: Razvan Pascanu's avatar Razvan Pascanu

R operator for specify shape

上级 e998600e
......@@ -1782,6 +1782,9 @@ class SpecifyShape(Op):
return [gz, None]
return [specify_shape(gz,s), None]
def R_op(self, inputs, eval_points):
return [zeros_like(x) for x in self.make_node(*inputs).outputs ]
specify_shape = SpecifyShape()
class MaxAndArgmax(Op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论