提交 1359eee8 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

R operator for Flatten op.

上级 83ac43d3
......@@ -4431,6 +4431,11 @@ class Flatten(Op):
g_out, = grads
return [reshape(g_out, shape(x), x.ndim)]
def R_op(self, inputs, eval_points):
if None in eval_points:
return [None]
return self.make_node(*eval_points).outputs
def flatten(x, outdim=1):
return Flatten(outdim)(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论