ó àÆ÷Xc@`sgddlmZmZmZddlZddlZddlmZdefd„ƒYZ d„Z dS(i(tabsolute_importtprint_functiontdivisionN(tModet MonitorModecB`sMeZdZddddd„Zd„Zd„Zd„Zddd„ZRS( s `MonitorMode` is a debug mode to easily step through function execution. Its default behavior is to behave like the 'FAST_RUN' mode. By providing either a `pre_func` (called before a node is executed) or a `post_func` (called after a node is executed) monitoring function, the user can inspect node behavior. A typical use case is to detect the introduction of NaN values in a graph. For an example of such a use case, see doc/tutorial/debug_faq.txt. Parameters ---------- pre_func A function to call before executing a thunk, with arguments: - the thunk index - the Apply node - the thunk to be called post_func A function to call after executing a thunk, with the same three arguments as `pre_func`. optimizer The optimizer to use. One may use for instance 'fast_compile' to skip optimizations. linker DO NOT USE. This mode uses its own linker. The parameter is needed to allow selecting optimizers to use. tdefaultcC`s¥||_||_tjjtjjƒg|jgƒ}|dkrTtjj}n|dk r…t |j t ƒ r…t d|ƒ‚ntt |ƒj|d|ƒdS(NRsDMonitorMode can only use its own linker! You should not provide one.t optimizer(tpre_funct post_functtheanotgoftWrapLinkerManyt OpWiseCLinkertevaltconfigRtNonet isinstancetmodeRt Exceptiontsupert__init__(tselfRRRtlinkert wrap_linker((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pyR)s     cC`s1tt|ƒjƒ\}}|||j|jfS(N(RRt __getstate__RR(Rtlnktopt((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pyR8scC`sD|\}}}}||_||_tt|ƒj||fƒdS(N(RRRRt __setstate__(RtstateRRRR((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pyR<s  cC`sU|jdk r%|j|||ƒn|ƒ|jdk rQ|j|||ƒndS(s8 The method that calls the thunk `fn`. N(RRR(Rtitnodetfn((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pyR Bs tc K`sI|dkr|j}nt|ƒd|jd|jddd|ƒ}|S(sÄ Create a new instance of this Mode. Keyword arguments can be provided for the linker, but they will be ignored, because MonitorMode needs to use its own linker. R RRRRN(tprovided_optimizerttypeRRR(Rt link_kwargsRtkwargstnew_mode((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pytcloneMs    N( t__name__t __module__t__doc__RRRRR R&(((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pyR s   cC`s¹x²|jD]§}t|dtjjƒ r tj|dƒjƒr tdƒtj j |ƒtdg|j D]}|d^qmƒtdg|jD]}|d^q•ƒPq q WdS(Nis*** NaN detected ***s Inputs : %ss Outputs: %s( toutputsRtnptrandomt RandomStatetisnantanytprintR tprintingt debugprinttinputs(RRRtoutputtinput((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pyt detect_nan^s ((( t __future__RRRtnumpyR+R ttheano.compile.modeRRR6(((s:/tmp/pip-build-X4mzal/theano/theano/compile/monitormode.pyts   T