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