提交 7eb3caf2 authored 作者: Frederic Bastien's avatar Frederic Bastien

complete the pydotprint change that add the input number as label to edge.

上级 a5256ba2
......@@ -410,6 +410,8 @@ def pydotprint(fct, outfile=os.path.join(config.compiledir,'theano.pydotprint.pn
for id,var in enumerate(node.inputs):
varstr=var_name(var)
label=''
if len(node.inputs)>1:
label=str(id)
if var.owner is None:
g.add_node(pd.Node(varstr,color='green'))
g.add_edge(pd.Edge(varstr,astr, label=label))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论