+ jI!* a0t$^RIHt^RIt^RIHt^RIHt^RIH t ^RI H t ^RI H t ^RIt^RIHt]!R 4t]'Ed]P$t]P(t^R IHt^R IHt^R IHt]!]R R4t]f ]P:t/]P<]P>b]P@]PBb]PD]PFb]PH]PJb]PL]PNb]PP]PPb]PR]PRb]PT]PTb]PV]PVb]PX]PXb]PZ]P\b]P^]P^b]P`]Pbb]Pd]Pdb]Pf]Phb]Pj]Plb]Pn]Pnb/]Pp]Prb]Pt]Ptb]Pv]Pxb]Pz]Pzb]P|]P~b]P]Pb]P]Pb]P]Pb]P]Pb]P]Pb]P]Pb]P]Pb]P]Pb]P]Pb]P]Pb]P]Pb]P]PbC]P]P]P]P]P]P]P]P]P]P]P]P]P]P]PPPR]P]!]P>4]P]!]PB4]P]P]P]P]P]P]P]!R^^.4/Ctp!RR] 4tq/trR]sR&RRlttRRltuRRRRRRR/RlltvR#)) annotationsN)Any) import_module)Printer)sympy_deprecation_warning) is_sequence)partialaesara)nlinalg)Elemwise) DimShuffle true_dividec.\P^^4#r)aetcomplexZ/home/luca/screen-checking/.venv/lib/python3.14/site-packages/sympy/printing/aesaracode.pyrGsS[[15Erca]tRt^RtRtRtV3RltRRltRRltRt Rt Rt R t R t R t];ttR tR tRtRtRtRtRtRtRtRtRtRtRRltRtV;t#) AesaraPrintera  .. deprecated:: 1.14. The ``Aesara Code printing`` is deprecated.See its documentation for more information. See :ref:`deprecated-aesaraprinter` for details. Code printer which creates Aesara symbolic expression graphs. Parameters ========== cache : dict Cache dictionary to use. If None (default) will use the global cache. To create a printer which does not depend on or alter global state pass an empty dictionary. Note: the dictionary is not copied on initialization of the printer and will be updated in-place, so using the same dict object when creating multiple printers or making multiple calls to :func:`.aesara_code` or :func:`.aesara_function` means the cache is shared between all these applications. Attributes ========== cache : dict A cache of Aesara variables which have been created for SymPy symbol-like objects (e.g. :class:`sympy.core.symbol.Symbol` or :class:`sympy.matrices.expressions.MatrixSymbol`). This is used to ensure that all references to a given symbol in an expression (or multiple expressions) are printed as the same Aesara variable, which is created only once. Symbols are differentiated only by name and type. The format of the cache's contents should be considered opaque to the user. _aesarac V<VPR/4Vn\SV` !V/VBR#)cacheN)poprsuper__init__)selfargskwargs __class__s&*,rrAesaraPrinter.__init__ts'ZZ,  $)&)rc TVf VPpV\V4VPW43#)zGet the cache key for a SymPy object. Parameters ========== s : sympy.core.basic.Basic SymPy object to get key for. name : str Name of object, if it does not have a ``name`` attribute. )nametyper )rsr%dtype broadcastables&&&&&r_get_keyAesaraPrinter._get_keyxs) <66Dd1gqvvu<))D)3KK.v.)D8}EsAc 4\VP44#rC)floatevalf)rnr!s&&,r _print_NumberAesaraPrinter._print_NumbersQWWYrc jVPR/4PV4pVPWRR7#)r6r.NNr8)rXr!r(s&&, r_print_MatrixSymbol!AesaraPrinter._print_MatrixSymbols3 8R(,,Q/""1"NNrc \\R4'g \R4h\PVP 4UUu.uF%pVUu.uFq@P !V3/VBNK upNK' upp4#uupiuuppi) stacklistsz>Matrix translation not yet supported in this version of Aesara)hasattrrNotImplementedErrorrXtolistrE)rrTr!LrHs&&, r_print_DenseMatrix AesaraPrinter._print_DenseMatrixsxsL))%OQ Q~~XXZ 45 51C[[ ' '1 5   5 s A< A7(A< 7A< c VPUu.uFq0P!V3/VBNK ppV^,pVR,Fp\PWV4pK V#uupi)rNN)r rErdot)rrFr!rHrIresultchilds&&, r _print_MatMulAesaraPrinter._print_MatMulsX:>))D)3KK.v.)D!b\\EWWV+F" EsAc @VPUu.uFq0P!V3/VBNK pp^p\V^,\4'dFV^,^8d8\ V^,4Fp\ P WT^,4pK! V#\R4huupi)raz^Only non-negative integer powers of matrices can be handled by Aesara at the moment)r rE isinstanceintrangerrbrZ)rrFr!rHrIrcis&&, r _print_MatPowAesaraPrinter._print_MatPows:>))D)3KK.v.)D hqk3 ' 'HQK!O8A;'!5(  &'HI I EsBc VP!VP3/VBpVP!\VP!3/VBpVP!\VP!3/VBpW4V3,#rC)rEparentslicerowslicecolslice)rrFr!rorqrrs&&, r_print_MatrixSlice AesaraPrinter._print_MatrixSlices_T[[3F3;;udmm4??;;udmm4??())rc ~VPPwr4\V4UUu.uFAp\V4Uu.uF(pVP!VPWV3,3/VBNK* upNKC ppp\P !^.VUu.uFp\P !^.VO5!NK upO5!#uupiuuppiuupir)blocksr/rjrErjoin) rrFr!nrowsncolsrcrvrows &&, r_print_BlockMatrix AesaraPrinter._print_BlockMatrixs{{(( "'u/!-A"'u/!-A;;t{{140;F;!-/!- /xxB&A&3SXXa.#.&ABB//BsB4.B/ 'B4!B: /B4c \VPVPVP3Uu.uF8p\ V\ P 4'dVP!V3/VBMTNK: up!#uupirC)rpstartstopsteprhsympyBasicrE)rrFr!rks&&, r _print_sliceAesaraPrinter._print_slicesp"&**dii!CE!CA&a55{{1//;<=!CEF FEs>A.c "\P#rC)mathpirrFr!s&&,r _print_PiAesaraPrinter._print_Pis wwrc ^RIpVP^,PwrEVP!V3/VBpVP!V3/VBp\VP4^8Xd!\P WgVP 4#VP!\P!VPR,!3/VBp\P WgV4#)rNr`) numpyr rElenrswitchnanr Piecewise) rrFr!npecondp_condp_e p_remainings &&, r_print_PiecewiseAesaraPrinter._print_Piecewises))A,##T,V,kk!&v& tyy>Q ::f2662 2kk%//499R="ALVL zz&{33rc \VP!VP3/VBVP!VP3/VB4#rC)rrEpqrs&&,r_print_RationalAesaraPrinter._print_Rationals84;;tvv88;;tvv88: :rc VP#rC)rrs&&,r_print_IntegerAesaraPrinter._print_Integers vv rc VP!\P!VP^,^,43/VB#r)rErgammar rs&&,r_print_factorialAesaraPrinter._print_factorials,{{5;;tyy|a'78CFCCrc ^RIHpVP!VP3/VBpVPF2pVP!V3/VBpV!WE\ P V44pK4 V#)r)Rop)aesara.gradientrrErF variablesr ones_like)rderivr!rrvvars&&, r_print_DerivativeAesaraPrinter._print_DerivativesY' [[ .v .??C++c,V,CRcmmC01B# rc V#rCr)rrFs&&r emptyPrinterAesaraPrinter.emptyPrinters rc @Vf/pVf/pVPWVR7#)aXConvert a SymPy expression to a Aesara graph variable. The ``dtypes`` and ``broadcastables`` arguments are used to specify the data type, dimension, and broadcasting behavior of the Aesara variables corresponding to the free symbols in ``expr``. Each is a mapping from SymPy symbols to the value of the corresponding argument to ``aesara.tensor.var.TensorVariable``. See the corresponding `documentation page`__ for more information on broadcasting in Aesara. .. __: https://aesara.readthedocs.io/en/latest/reference/tensor/shapes.html#broadcasting Parameters ========== expr : sympy.core.expr.Expr SymPy expression to print. dtypes : dict Mapping from SymPy symbols to Aesara datatypes to use when creating new Aesara variables for those symbols. Corresponds to the ``dtype`` argument to ``aesara.tensor.var.TensorVariable``. Defaults to ``'floatX'`` for symbols not included in the mapping. broadcastables : dict Mapping from SymPy symbols to the value of the ``broadcastable`` argument to ``aesara.tensor.var.TensorVariable`` to use when creating Aesara variables for those symbols. Defaults to the empty tuple for symbols not included in the mapping (resulting in a scalar). Returns ======= aesara.graph.basic.Variable A variable corresponding to the expression's value in a Aesara symbolic expression graph. )r6r7)rE)rrFr6r7s&&&&rdoprintAesaraPrinter.doprints-R >F  !N{{4~{NNr)rNNNrS) __name__ __module__ __qualname____firstlineno____doc__ printmethodrr*r3r;r@rJrPrUr]_print_ImmutableMatrix_print_ImmutableDenseMatrixrerlrsr}rrrrrrrrr__static_attributes__ __classcell__)r"s@rrrRs>K*=$.E P   O `). Defaults to the module-level global cache. dtypes : dict Passed to :meth:`.AesaraPrinter.doprint`. broadcastables : dict Passed to :meth:`.AesaraPrinter.doprint`. Returns ======= aesara.graph.basic.Variable A variable corresponding to the expression's value in a Aesara symbolic expression graph. z9 The aesara_code function is deprecated. 1.14deprecated-aesaraprinterdeprecated_since_versionactive_deprecations_targetz"aesara is required for aesara_code)rsettings)rr ImportErrorrrr)rFrr!s&&,r aesara_coderHsS8 "(#=  6>?? } ur 2 : :4 J6 JJrc Ve\PVRV,4#VeZ\VP44pVP 4UUu/uF#wrVVRV,RWF, ,,bK% upp#VeV#/#uuppi)a Get value of ``broadcastables`` argument to :func:`.aesara_code` from keyword arguments to :func:`.aesara_function`. Included for backwards compatibility. Parameters ========== inputs Sequence of input symbols. dim : int Common number of dimensions for all inputs. Overrides other arguments if given. dims : dict Mapping from input symbols to number of dimensions. Overrides ``broadcastables`` argument if given. broadcastables : dict Explicit value of ``broadcastables`` argument to :meth:`.AesaraPrinter.doprint`. If not None function will return this value unchanged. Returns ======= dict Dictionary mapping elements of ``inputs`` to their "broadcastable" values (tuple of ``bool``\ s). F)T)dictfromkeysmaxvaluesitems)inputsdimdimsr7maxdimr'ds&&&& r dim_handlingrus> }}VX^44 T[[]#  $ x!|g44 4$  ! I s)Brrr7c Zaa\RRRR7\'g \R4hVPR/4pVPR/4p\ WWER7p\ \ WxVR 7p \\W44p \\W44p V U u.uFNp \V \PPP4'dT M\PV 4NKP p p \V 4^8Xd V ^,p \P !W3/VBoSP"U u.uF%p \V P$P&4^8HNK' up oV'd\)S4'g SSnS#VV3R lpSVnSP.VnSVnV#uup iuup i) ao Create a Aesara function from SymPy expressions. The inputs and outputs are converted to Aesara variables using :func:`.aesara_code` and then passed to ``aesara.function``. Parameters ========== inputs Sequence of symbols which constitute the inputs of the function. outputs Sequence of expressions which constitute the outputs(s) of the function. The free symbols of each expression must be a subset of ``inputs``. scalar : bool Convert 0-dimensional arrays in output to scalars. This will return a Python wrapper function around the Aesara function object. cache : dict Cached Aesara variables (see :class:`AesaraPrinter.cache `). Defaults to the module-level global cache. dtypes : dict Passed to :meth:`.AesaraPrinter.doprint`. broadcastables : dict Passed to :meth:`.AesaraPrinter.doprint`. dims : dict Alternative to ``broadcastables`` argument. Mapping from elements of ``inputs`` to integers indicating the dimension of their associated arrays/tensors. Overrides ``broadcastables`` argument if given. dim : int Another alternative to the ``broadcastables`` argument. Common number of dimensions to use for all arrays/tensors. ``aesara_function([x, y], [...], dim=2)`` is equivalent to using ``broadcastables={x: (False, False), y: (False, False)}``. Returns ======= callable A callable object which takes values of ``inputs`` as positional arguments and returns an output array for each of the expressions in ``outputs``. If ``outputs`` is a single expression the function will return a Numpy array, if it is a list of multiple expressions the function will return a list of arrays. See description of the ``squeeze`` argument above for the behavior when a single output is passed in a list. The returned object will either be an instance of ``aesara.compile.function.types.Function`` or a Python wrapper function around one. In both cases, the returned value will have a ``aesara_function`` attribute which points to the return value of ``aesara.function``. Examples ======== >>> from sympy.abc import x, y, z >>> from sympy.printing.aesaracode import aesara_function A simple function with one input and one output: >>> f1 = aesara_function([x], [x**2 - 1], scalar=True) >>> f1(3) 8.0 A function with multiple inputs and one output: >>> f2 = aesara_function([x, y, z], [(x**z + y**z)**(1/z)], scalar=True) >>> f2(3, 4, 2) 5.0 A function with multiple inputs and multiple outputs: >>> f3 = aesara_function([x, y], [x**2 + y**2, x**2 - y**2], scalar=True) >>> f3(2, 3) [13.0, -5.0] See also ======== dim_handling z= The aesara_function function is deprecated. rrrz&Aesara is required for aesara_functionrr6)rrr7)rr6r7c<S!V!p\V4'd8\V4UUu.uFwr#SV,'d VR,MTNK! upp#VR,#uuppi)Nr)r enumerate)r outrkofuncis_0ds* rwrapper aesara_function..wrapper'sTDk s  9B3HU1XXAbE1,H Hr7NIs%A)rr rrrr rlistmaprhgraphbasicVariableras_tensor_variablerfunctionoutputsvariabler)anyaesara_function __wrapped__r)rrscalarrrr7r!rr6codetinputstoutputsoutputrrrrs&&&$$$, @@rrrsr "(#=  6BCC JJw #E ZZ" %F!dN ;e"0 2D3t$%GC&'H}EF}Erx*VV\\-?-?-H-HIIsOeOeflOmm}EHF 8}A; ??7 7 7D9= FAS)) *a / FE U# GllGO"G N9F Gs AF#++F()FFrCrr)w__conditional_annotations__ __future__rrtypingrsympy.externalrsympy.printing.printerrsympy.utilities.exceptionsrsympy.utilities.iterablesrr functoolsr r raesr0r aesara.tensorr aesara.tensor.elemwiser r getattrrtrue_divAddaddMulmulAbsabssignsgnceilingceilfloorlogexpsqrtcosacosarccossinasinarcsintanatanarctanatan2arctan2coshacosharccoshsinhasinharcsinhtanhatanharctanhrerealimimagrHangleerfrloggammagammalnPowpowEqeqStrictGreaterThangtStrictLessThanltLessThanle GreaterThangeAnd bitwise_andOr bitwise_orNotinvertXor bitwise_xorMaxmaximumMinminimum conjugateconjcorenumbers ImaginaryUnitMatAddHadamardProductTracetrace DeterminantdetInversematrix_inverse TransposerDrr__annotations__rrr)rs@rrQsQ"" (*@1  x  6 --C --C%/1 #}d3Kll 2 IIsww2 IIsww2 IIsww2 JJ 2 MM388 2 KK 2 IIsww2 IIsww2 JJ2 IIsww2 JJ 2 IIsww2 JJ 2 IIsww2 JJ 2 KK!2" JJ#2$ KK%2& JJ'2( KK)2* JJ+2, KK-2. HHchh/20 HHchh122 IIsyy324 IIsww526 KK728 NNCKK92: IIsww;2< HHcff=2>  # #SVV?2@  #&&A2B NNCFFC2D   svvE2F IIs HHcnn IIszz IIs IIs{{ IIs{{ OOSXX JJ   , ,.E LL(377+  ! !8CGG#4 KK    MM711 OOZA?c2GjpOGpOf " n!*KZ,^O OO,0Or