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

Modified the printing of constants in pydotprint to include their type

上级 0bcfafe6
...@@ -372,7 +372,7 @@ def pydotprint(fct, outfile=os.path.join(config.compiledir,'theano.pydotprint.pn ...@@ -372,7 +372,7 @@ 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 = str(var.data) varstr = '%s [%s]'% (str(var.data) , 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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论