提交 d1879e8b authored 作者: Ian Goodfellow's avatar Ian Goodfellow

corrected SpecifyShape.grad

上级 e205110f
...@@ -2204,8 +2204,8 @@ class SpecifyShape(Op): ...@@ -2204,8 +2204,8 @@ class SpecifyShape(Op):
# to remove that op from the graph to don't block other optimization # to remove that op from the graph to don't block other optimization
# Should I do an optimizer that will remove the SpecifyShape? # Should I do an optimizer that will remove the SpecifyShape?
# I think Yes # I think Yes
return [gz, None] return [gz, DisconnectedType()()]
return [specify_shape(gz, s), None] return [specify_shape(gz, s), DisconnectedType()()]
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
if eval_points[0] is None: if eval_points[0] is None:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论