提交 3315723a authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3625 from sieben/dict_litteral

Dict can be litteral
...@@ -177,10 +177,9 @@ class PyDotFormatter(object): ...@@ -177,10 +177,9 @@ class PyDotFormatter(object):
for id, var in enumerate(node.inputs): for id, var in enumerate(node.inputs):
var_id = self.__node_id(var.owner if var.owner else var) var_id = self.__node_id(var.owner if var.owner else var)
if var.owner is None: if var.owner is None:
vparams = {} vparams = {'name': var_id,
vparams['name'] = var_id 'label': var_label(var),
vparams['label'] = var_label(var) 'node_type': 'input'}
vparams['node_type'] = 'input'
if isinstance(var, gof.Constant): if isinstance(var, gof.Constant):
vparams['node_type'] = 'constant_input' vparams['node_type'] = 'constant_input'
elif isinstance(var, theano.tensor.sharedvar. elif isinstance(var, theano.tensor.sharedvar.
...@@ -217,13 +216,12 @@ class PyDotFormatter(object): ...@@ -217,13 +216,12 @@ class PyDotFormatter(object):
var_id = self.__node_id(var) var_id = self.__node_id(var)
if var in outputs or len(var.clients) == 0: if var in outputs or len(var.clients) == 0:
vparams = {} vparams = {'name': var_id,
vparams['name'] = var_id 'label': var_label(var),
vparams['label'] = var_label(var) 'node_type': 'output',
vparams['node_type'] = 'output' 'dtype': type_to_str(var.type),
vparams['dtype'] = type_to_str(var.type) 'tag': var_tag(var),
vparams['tag'] = var_tag(var) 'style': 'filled'}
vparams['style'] = 'filled'
if len(var.clients) == 0: if len(var.clients) == 0:
vparams['fillcolor'] = self.node_colors['unused'] vparams['fillcolor'] = self.node_colors['unused']
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论