o 3Ih@s"ddlmZmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZmZdd lmZmZd d d d ZejdZeeZd d ddZd d ddZd d ddZGdddeZGdddeZGdddeZGdddeZGdddeZ Gdd d eZ!d!S)")AddArgumentIndexErrorFunction)Pow)S)default_sort_key)sympify)explog)MaxMin)TokennoneTevaluatecC"ttt||dt||d|dSNr)r rr x1x2rrY/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/sympy/codegen/numpy_nodes.py _logaddexp s"rcCst||dtSr)r _ln2xrrrr_lbsrcCstt||dSr)r_tworrrr_exp2sr cCrr)rrr rrrr _logaddexp2s r!c@>eZdZdZdZddZdddZdd Zd d Zd d Z dS) logaddexpz Logarithm of the sum of exponentiations of the inputs. Helper class for use with e.g. numpy.logaddexp See Also ======== https://numpy.org/doc/stable/reference/generated/numpy.logaddexp.html rcGtj|gt|tdRSN)keyr__new__sortedrclsargsrrrr(,zlogaddexp.__new__r cCJ|dkr |j\}}n|dkr|j\}}nt||tjtjt||Sz@ Returns the first derivative of this function. r r)r,rrOner selfargindexwrtotherrrrfdiff/    zlogaddexp.fdiffcK t||SN)rr2rrkwargsrrr_eval_rewrite_as_log; zlogaddexp._eval_rewrite_as_logcO|tj|i|Sr9rewriter evalfr2r,r;rrr _eval_evalf>zlogaddexp._eval_evalfcBfdd|jD\}}t||}|t||ddkr|St||S)Nc3s |] }|jdiVqdSNr)simplify.0rr;rr Bsz+logaddexp._eval_simplify..Fr)r,rr#r2r,r;ab candidaterrJr_eval_simplifyA   zlogaddexp._eval_simplifyNr __name__ __module__ __qualname____doc__nargsr(r6r<rCrPrrrrr#     r#c@r") logaddexp2z Logarithm of the sum of exponentiations of the inputs in base-2. Helper class for use with e.g. numpy.logaddexp2 See Also ======== https://numpy.org/doc/stable/reference/generated/numpy.logaddexp2.html rcGr$r%r'r*rrrr(Vr-zlogaddexp2.__new__r cCr.r/)r,rrr0r r1rrrr6Yr7zlogaddexp2.fdiffcKr8r9)r!r:rrrr<er=zlogaddexp2._eval_rewrite_as_logcOr>r9r?rBrrrrChrDzlogaddexp2._eval_evalfcrE)Nc3s$|] }|jdiVqdSrF)rGfactorrHrJrrrKls"z,logaddexp2._eval_simplify..Fr)r,r!rZrLrrJrrPkrQzlogaddexp2._eval_simplifyNrRrSrrrrrZJrYrZc@(eZdZdZdZZdeiZee Z dS)aminz Minimum value along an axis. Helper class for use with e.g. numpy.amin See Also ======== https://numpy.org/doc/stable/reference/generated/numpy.amin.html arrayaxisr`N rTrUrVrW __slots___fieldsrdefaults staticmethodr_construct_axisrrrrr]t   r]c@r\)amaxz Maximum value along an axis. Helper class for use with e.g. numpy.amax See Also ======== https://numpy.org/doc/stable/reference/generated/numpy.amax.html r^r`Nrarrrrrhrgrhc@eZdZdZddZdS)maximumz Element-wise maximum of array elements. Helper class for use with e.g. numpy.maximum See Also ======== https://numpy.org/doc/stable/reference/generated/numpy.maximum.html cG t|jSr9)r r,r2r,rrr_eval_rewrite_as_Maxr=zmaximum._eval_rewrite_as_MaxN)rTrUrVrWrmrrrrrj rjc@ri)minimumz Element-wise minimum of array elements. Helper class for use with e.g. numpy.minimum See Also ======== https://numpy.org/doc/stable/reference/generated/numpy.minimum.html cGrkr9)r r,rlrrr_eval_rewrite_as_Minr=zminimum._eval_rewrite_as_MinN)rTrUrVrWrprrrrrornroN)"sympy.core.functionrrrsympy.core.powerrsympy.core.singletonrsympy.core.sortingrsympy.core.sympifyr&sympy.functions.elementary.exponentialr r (sympy.functions.elementary.miscellaneousr r astrrrr0rrrr r!r#rZr]rhrjrorrrrs(     **