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

rearranged _traverse to be easier to read

上级 d9df2be5
...@@ -213,9 +213,10 @@ def Rop(f, wrt, eval_points): ...@@ -213,9 +213,10 @@ def Rop(f, wrt, eval_points):
def _traverse(node): def _traverse(node):
""" TODO: writeme """ """ TODO: writeme """
if node is None: if node is None:
return None return
else:
op = node.op op = node.op
inputs = node.inputs inputs = node.inputs
...@@ -273,7 +274,7 @@ def Rop(f, wrt, eval_points): ...@@ -273,7 +274,7 @@ def Rop(f, wrt, eval_points):
same_type_eval_points.append(y) same_type_eval_points.append(y)
seen_nodes[node] = op.R_op(node.inputs, same_type_eval_points) seen_nodes[node] = op.R_op(node.inputs, same_type_eval_points)
return None #end _traverse
# Populate the dictionary # Populate the dictionary
for out in f: for out in f:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论