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

Fixed pydotprint when constant data spans multiple lines.

上级 efc5e1c0
...@@ -376,7 +376,12 @@ def pydotprint(fct, outfile=os.path.join(config.compiledir,'theano.pydotprint.pn ...@@ -376,7 +376,12 @@ def pydotprint(fct, outfile=os.path.join(config.compiledir,'theano.pydotprint.pn
if var.name is not None: if var.name is not None:
varstr = var.name varstr = var.name
elif isinstance(var,gof.Constant): elif isinstance(var,gof.Constant):
varstr = '%s [%s]'% (str(var.data) , str(var.type)) dstr = str(var.data)
if '\n' in dstr:
dstr = dstr[:dstr.index('\n')]
if len(dstr) > 30:
dstr = dstr[:27]+'...'
varstr = '%s [%s]'% (dstr, str(var.type))
elif var in input_update and input_update[var].variable.name is not None: elif var in input_update and input_update[var].variable.name is not None:
varstr = input_update[var].variable.name varstr = input_update[var].variable.name
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论