提交 cb86fda4 authored 作者: James Bergstra's avatar James Bergstra

DebugMode - added vertical lines to debugprint to help (a bit) to follow

indentation in big graphs... still a bit ugly though.
上级 8c8105f1
...@@ -354,14 +354,17 @@ def debugprint(r, prefix='', depth=-1, done=None, file=sys.stdout): ...@@ -354,14 +354,17 @@ def debugprint(r, prefix='', depth=-1, done=None, file=sys.stdout):
# this variable is the output of computation, # this variable is the output of computation,
# so just print out the apply # so just print out the apply
a = r.owner a = r.owner
print >> file, prefix, a.op, id(a) if len(a.outputs) == 1:
print >> file, '%s%s [@%i]' % (prefix, a.op, id(r))
else:
print >> file, '%s%s.%i [@%i]' % (prefix, a.op, a.outputs.index(r), id(r))
if id(a) not in done: if id(a) not in done:
done.add(id(a)) done.add(id(a))
for i in a.inputs: for i in a.inputs:
debugprint(i, prefix+' ', depth=depth-1, done=done, file=file) debugprint(i, prefix+' |', depth=depth-1, done=done, file=file)
else: else:
#this is a variable #this is a variable
print >> file, prefix, r, id(r) print >> file, '%s%s [@%i]' % (prefix, r, id(r))
return file return file
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论