提交 3afc64ab authored 作者: James Bergstra's avatar James Bergstra

added debugprints

上级 4591855f
......@@ -7,7 +7,7 @@ from theano.configparser import config, AddConfigVar, StrParam
from theano.gof import (utils, Op, view_roots, PatternSub, DestroyHandler,
SeqOptimizer, local_optimizer, Optimizer, LocalOptimizer, OpKeyOptimizer,
InconsistencyError, toolbox, SequenceDB, EquilibriumOptimizer)
from theano.printing import pprint, FunctionPrinter
from theano.printing import pprint, FunctionPrinter, debugprint
from theano.compile.mode import optdb
from theano.gof.python25 import any
import theano.scalar
......@@ -1063,3 +1063,8 @@ blas_optdb.register('local_dot22_to_dot22scalar',
11, 'fast_run')
#@opt.register_stabilize
@gof.local_optimizer([])
def local_print_as_we_go_along(node):
if node.op == tensor.true_div:
debugprint(node)
......@@ -8,7 +8,7 @@ from theano import gof
from theano import scalar
from theano import printing
from theano.tensor import basic as tensor
from theano.printing import pprint
from theano.printing import pprint, debugprint
from theano.tensor import elemwise
from theano.tensor import opt
from theano.compile import optdb
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论