提交 b78da17b authored 作者: Frederic's avatar Frederic

backport to python 2.4

上级 6171e71f
...@@ -926,9 +926,13 @@ eig = Eig() ...@@ -926,9 +926,13 @@ eig = Eig()
def _zero_disconnected(outputs, grads): def _zero_disconnected(outputs, grads):
return [o.zeros_like() l = []
if isinstance(g.type, DisconnectedType) else g for o, g in zip(outputs, grads):
for o, g in zip(outputs, grads)] if isinstance(g.type, DisconnectedType):
l.append(o.zeros_like())
else:
l.append(g)
return l
class Eigh(Eig): class Eigh(Eig):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论