提交 122a7597 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix partial commit.

上级 a96f4908
......@@ -409,6 +409,7 @@ def pydotprint(fct, outfile=os.path.join(config.compiledir,'theano.pydotprint.pn
g.add_node(pd.Node(astr,shape='box'))
for id,var in enumerate(node.inputs):
varstr=var_name(var)
label=''
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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论