B }[R!\@s<ddlmZmZddlZddlZddlmZddlmZddl m Z ddl Z ddl m Z edZerejZejZddlmZe jeje jeje jeje jeje jeje j ej e j!ej!e j"ej"e j#ej#e j$ej$e j%ej&e j'ej'e j(ej)e j*ej*e j+ej,e j-ej.e j/ej/e j0ej1e j2ej2e j3ej4e j5ej5e j6ej7e j8ej9e j:ej;e jej>e j?ej?e j@ejAe jBejCe jDejEe jFejGe jHejIe jJejKe jLejMe jNejOe jPejQe jRejSe jTejUe jVeWeje jXeWeje jYejZe j[ej\e j]ej^e j_e`d d dgi,ZaGd d d eZbiZcecfd dZddiidfddZeidfddZfdS))print_functiondivisionN) import_module)Printer)range)partialtheano)linalg)FFcseZdZdZdZfddZiifddZiifddZd d Zd d Z ifd dZ ddZ e Z Z ddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(ZZS)) TheanoPrinterz& Code printer for Theano computations Z_theanocs&|dt|_tt|j||dS)Ncache)popdictr superr __init__)selfargskwargs) __class__8lib/python3.7/site-packages/sympy/printing/theanocode.pyrFszTheanoPrinter.__init__cCsb||d}||d}|j||t|f}||jkr>|j|Stj|j||d}||j|<|SdS)NfloatXr)namedtype broadcastable)getrtyper tttensor)rsdtypesbroadcastablesrrkeyvaluerrr _print_SymbolJs     zTheanoPrinter._print_Symbolc Cs||d}||d}tt|dt|jd}|||t||jf}||jkr^|j|Stj|||d}||j|<|SdS)Nrr_r)rrr)rstrrrr rr) rrr r!rrrr"r#rrr_print_AppliedUndefUs     z!TheanoPrinter._print_AppliedUndefc s*tt|}fdd|jD}||S)Ncsg|]}j|fqSr)_print).0arg)rrrr dsz.TheanoPrinter._print_Basic..)mappingrr)rexprropchildrenr)rrr _print_Basicbs zTheanoPrinter._print_BasiccKs tt|S)N)evalr&)rnrrrr _print_NumbergszTheanoPrinter._print_NumbercKsT||d}|j|t|f}||jkr0|j|St|d|j}||j|<|SdS)Nr)FF)rrrr rZTensor)rXr rrr"r#rrr_print_MatrixSymboljs    z!TheanoPrinter._print_MatrixSymbolc sJy tjWntk r&tdYn Xtfdd|DSdS)Nz>Matrix translation not yet supported in this version of Theanocs g|]}fdd|DqS)csg|]}j|fqSr)r()r)r*)rrrrr+{sz?TheanoPrinter._print_DenseMatrix...r)r)L)rrrrr+{sz4TheanoPrinter._print_DenseMatrix..)rZ stacklistsAttributeErrorNotImplementedErrortolist)rr4rr)rrr_print_DenseMatrixts  z TheanoPrinter._print_DenseMatrixc sDfdd|jD}|d}x |ddD]}t||}q,W|S)Ncsg|]}j|fqSr)r()r)r*)rrrrr+sz/TheanoPrinter._print_MatMul..rr )rrdot)rr-rr/resultZchildr)rrr _print_MatMuls zTheanoPrinter._print_MatMulcKsD|j|jf|}|jt|jf|}|jt|jf|}|||fS)N)r(parentslicerowslicecolslice)rr-rr>r@rArrr_print_MatrixSlicesz TheanoPrinter._print_MatrixSlicec s@jj\}fddt|D}tjddd|DS)Ncs(g|] fddtDqS)cs$g|]}jj|ffqSr)r(blocks)r)c)r-rrrrrr+sz?TheanoPrinter._print_BlockMatrix...)r)r))r-rncolsr)rErr+sz4TheanoPrinter._print_BlockMatrix..rcSsg|]}tjd|qS)r )r )rjoin)r)rowrrrr+s)r)rCshaperrrG)rr-rZnrowsrCr)r-rrFrr_print_BlockMatrixs  z TheanoPrinter._print_BlockMatrixc s$tfdd|j|j|jfDS)Ncs*g|]"}t|tjr"j|fn|qSr) isinstancesympyZBasicr()r)i)rrrrr+sz.TheanoPrinter._print_slice..)r?startstopstep)rr-rr)rrr _print_sliceszTheanoPrinter._print_slicecKsdS)Ng-DT! @r)rr-rrrr _print_PiszTheanoPrinter._print_Pic Ksddl}|jdj\}}t|jdkrJt|j|f||j|f||jSt|j|f||j|f||jtj|jddf|S)Nrr ) ZnumpyrlenrZswitchr(nanrLZ Piecewise)rr-rZnpeZcondrrr_print_Piecewises  zTheanoPrinter._print_PiecewisecKs$t|j|jf||j|jf|S)N)rZtrue_divr(pq)rr-rrrr_print_RationalszTheanoPrinter._print_RationalcKs|jS)N)rW)rr-rrrr_print_IntegerszTheanoPrinter._print_IntegercKs|jt|jddf|S)Nrr )r(rLgammar)rr-rrrr_print_factorialszTheanoPrinter._print_factorialcKsF|j|jf|}x0|jD]&}|j|f|}t||t|}qW|S)N)r(r-Z variablesrZRopZ ones_like)rZderivrrvvarrrr_print_Derivatives  zTheanoPrinter._print_DerivativecCs|S)Nr)rr-rrr emptyPrinterszTheanoPrinter.emptyPrintercKs|j|f|S)z7Returns printer's representation for expr (as a string))r()rr-rrrrdoprintszTheanoPrinter.doprint)__name__ __module__ __qualname____doc__Z printmethodrr$r'r0r3r5r:Z_print_ImmutableMatrixZ_print_ImmutableDenseMatrixr=rBrJrQrRrVrYrZr\r_r`ra __classcell__rr)rrr Bs,      r cKs"ts tdt|idj|f|S)Nz"theano is required for theano_code)r Zsettings)r ImportErrorr ra)r-r rrrr theano_codesrhrc sH|rtt||gt|rDttfdd|D}|S)zy Handle various input types for dimensions in tensor_wrap See Also: tensor_wrap theano_funciton c3s.|]&}|d|d|fVqdS))F)TNr)r)rM)dimsmaxdimrr szdim_handling..)rziprSmaxvalues)inputsZdimrir!keysrr)rirjr dim_handlings  rqc  sts td|dkrin|}t|f|}tjdkrBttdnttj }dd|Dt fdd| D}t t |||d }t t||} t t||} t| d kr| dn| } tj| | f|S) z/ Create Theano function from SymPy expressions z&theano is required for theano_functionN)rcSs g|]\}}|j|jkr|qSr)ZkindZPOSITIONAL_OR_KEYWORD)r)r2rWrrrr+sz#theano_function..c3s"|]\}}|kr||fVqdS)Nr)r)kv) dim_namesrrrksz"theano_function..)r r r!r )rrgrqsys version_infoinspectZ getargspecZ signatureZ parametersitemsrrrhlistmaprSZfunction) roZoutputsr r rr!ZparamZ theano_kwargscodeZtinputsZtoutputsr)rurtheano_functions  r})gZ __future__rrrxrvZsympy.externalrZsympy.printing.printerrZsympy.core.compatibilityrrL functoolsrrZscalarZtsrrZtheano.sandboxr ZtlinalgZAddaddZMulmulZAbsZabs_ZsignZsgnZceilingZceilZfloorlogZexpZsqrtZcosZacosZarccosZsinZasinZarcsinZtanZatanZarctanZatan2Zarctan2ZcoshZacoshZarccoshZsinhZasinhZarcsinhZtanhZatanhZarctanhrerealZimimagr*ZangleZerfr[ZloggammaZgammalnZPowpowZEqeqZStrictGreaterThangtZStrictLessThanltZLessThanleZ GreaterThangeZAndand_ZOror_ZMaxZmaximumZMinZminimumZMatAddZElemwiseZHadamardProductZTraceZtraceZ DeterminantZdetZInverseZmatrix_inverseZ TransposeZ DimShuffler,r Z global_cacherhrqr}rrrrsz     }