提交 c49edd05 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix pydotprint_variables.

上级 301305e8
...@@ -613,9 +613,13 @@ def pydotprint(fct, outfile=None, ...@@ -613,9 +613,13 @@ def pydotprint(fct, outfile=None,
def pydotprint_variables(vars, def pydotprint_variables(vars,
outfile=os.path.join(config.compiledir,'theano.pydotprint.png'), outfile=os.path.join(config.compiledir,'theano.pydotprint.png'),
depth = -1, depth = -1,
high_contrast = True): high_contrast = True, colorCodes = None):
''' Identical to pydotprint just that it starts from a variable instead ''' Identical to pydotprint just that it starts from a variable instead
of a compiled function. Could be useful ? ''' of a compiled function. Could be useful ? '''
if colorCodes is None:
colorCodes = default_colorCodes
try: try:
import pydot as pd import pydot as pd
except: except:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论