Sj??cFd|j|jddS)Nznot (r)_printrrs r _print_NotzExprPrinter._print_Nots#t{{499Q<0133rNordercN|j|jdt|S)Nz + r)rr rs r _print_AddzExprPrinter._print_Adds~~dii 40@AArcj|j|jd|jdt|S)N )rrrel_oprrs r_print_RelationalzExprPrinter._print_Relationals+~~dii1T[[M);Z=MNNrcJ|j|jdtdS)Nz & BitwiseAndrrrrs r_print_BitwiseFn_bitwise_andz(ExprPrinter._print_BitwiseFn_bitwise_and ~~dii <0HIIrcJ|j|jdtdS)Nz | BitwiseOrr%rs r_print_BitwiseFn_bitwise_orz'ExprPrinter._print_BitwiseFn_bitwise_or#s~~dii ;0GHHrcJ|j|jdtdS)Nz ^ BitwiseXorr%rs r_print_BitwiseFn_bitwise_xorz(ExprPrinter._print_BitwiseFn_bitwise_xor&r'rcP|j|jdtddz SN % Atom?r%rs r _print_ModzExprPrinter._print_Mod+#~~dii 60BS0HIIrc\|j|jdtddz }d|dS)N / r1r2(rr%)rr ss r_print_FloatTrueDivzExprPrinter._print_FloatTrueDiv.s/ NN499eZ-?#-E F1#Qxrc$|j|SN)_print_FloorDivrs r_print_CleanDivzExprPrinter._print_CleanDiv2s##D))rc>|j|jdSNrrrs r_print_IdentityzExprPrinter._print_Identity5s{{499Q<((rcv|jdk(r ttj|dSt|S)N5)dps)_precstrsympyFloatrs r _print_FloatzExprPrinter._print_Float9s2 :: u{{4R01 1t9 rc|j\}}|t|k7r t|t|}|dkrtd||dkDr|j|g|zdtdSy)Nrz#exponent must be non-negative, got rMul1)rintAssertionErrorrrrr baseexps r _print_PowzExprPrinter._print_PowIspII c #c(? % %#h 7 #Fse!LM M 7>>4&3,Z5FG Grc0tdt|)Nz#_print_ToFloat not implemented for NotImplementedErrortypers r_print_ToFloatzExprPrinter._print_ToFloatYs!$GT |"TUUrc0tdt|)Nz$_print_Infinity not implemented for rTrs r_print_InfinityzExprPrinter._print_Infinity\!$Hd "UVVrc0tdt|)Nz,_print_NegativeInfinity not implemented for rTrs r_print_NegativeInfinityz#ExprPrinter._print_NegativeInfinity_s!:4:, G  rc0tdt|)Nz_print_NaN not implemented for rTrs r _print_NaNzExprPrinter._print_NaNds!$CDJ<"PQQrc0tdt|)Nz$_print_FloorDiv not implemented for rTrs rr<zExprPrinter._print_FloorDivgrZrc0tdt|)Nz%_print_PythonMod not implemented for rTrs r_print_PythonModzExprPrinter._print_PythonModjs!$I$t*"VWWrc0tdt|)Nz&_print_IntTrueDiv not implemented for rTrs r_print_IntTrueDivzExprPrinter._print_IntTrueDivm!$J4PT:,"WXXrc0tdt|)N(_print_PowByNatural not implemented for rTrs r_print_PowByNaturalzExprPrinter._print_PowByNaturalp!6tDzl C  rc0tdt|)Nz$_print_FloatPow not implemented for rTrs r_print_FloatPowzExprPrinter._print_FloatPowurZrc0tdt|)Nz&_print_TruncToInt not implemented for rTrs r_print_TruncToIntzExprPrinter._print_TruncToIntxrdrc0tdt|)Nz&_print_RoundToInt not implemented for rTrs r_print_RoundToIntzExprPrinter._print_RoundToInt{rdrc0tdt|)Nz(_print_RoundDecimal not implemented for rTrs r_print_RoundDecimalzExprPrinter._print_RoundDecimal~rhrc0tdt|)Nz(_print_TruncToFloat not implemented for rTrs r_print_TruncToFloatzExprPrinter._print_TruncToFloatrhrr;)!__name__ __module__ __qualname__ printmethodrGExprrFrrrr"r&r*r-r3r9r=r@rIrRrWrYr\r^r<rarcrgrjrlrnrprrrrr r s#K@uzz@c@4uzz4c4BuzzB#*BBOejjOSOJJJI IsIJJJ JuzzJcJ s*EJJ*3*)EJJ)3) uzz c  V5::V#VWEJJW3W EJJ 3 RuzzRcRWEJJW3WXUZZXCXYejjYSY   s WEJJW3WYejjYSYYejjYSY   s    s rr ceZdZdejdefdZdejdefdZdejdefdZdejdefdZ dejdefdZ dejdefdZ dejdefd Z dejdefd Z dejdefd Zdejdefd Zdejdefd ZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZ dejdefdZ!dejdefdZ"dejdefd Z#dejdefd!Z$dejdefd"Z%dejdefd#Z&dejdefd$Z'dejdefd%Z(y&)' PythonPrinterr r ct|jdk7r tdd|j|jddS)N$ToFloat expects exactly one argumentztorch.sym_float(rrlenrrNrrs rrWzPythonPrinter._print_ToFloatsA tyy>Q  !GH H"$++diil";!>rcN|j|jdt|S)Nz and rrs r _print_AndzPythonPrinter._print_Ands~~dii*T2BCCrcN|j|jdt|S)Nz or rrs r _print_OrzPythonPrinter._print_Or~~diiD1ABBrcdfd|jD\}}}|dk7r d|d|d}d|d|dS)Nc3VK|] }j|tddz "ywr1r2N parenthesizer.0argrs r z7PythonPrinter._print_ModularIndexing..s- ADD  c:f#5#; < &)rLr7 // rr0rrr xdivmods` r_print_ModularIndexingz$PythonPrinter._print_ModularIndexingsN HL  3 #:A3d3%q!A1#SQrcy)Nzmath.infrxrs rrYzPythonPrinter._print_Infinityrcy)Nz -math.infrxrs rr\z%PythonPrinter._print_NegativeInfinitysrcy)Nzmath.nanrxrs rr^zPythonPrinter._print_NaNrrcP|j|jdtddz Sr/r%rs rrazPythonPrinter._print_PythonModr4rcBfd|jD\}}|d|S)Nc3VK|] }j|tddz "ywrrrs rrz0PythonPrinter._print_FloorDiv..s(Xs$##CF);c)ABXrrrrr rrs` rr<zPythonPrinter._print_FloorDivs%XdiiX3DrcP|j|jdtddz S)Nr6r1r2r%rs rrczPythonPrinter._print_IntTrueDivr4rc,d|j|dS)Nztorch._sym_sqrt(r)rrs r _helper_sqrtzPythonPrinter._helper_sqrts"$++d"3!4A66rc>|j|jdSr?)rrrs r_print_OpaqueUnaryFn_sqrtz'PythonPrinter._print_OpaqueUnaryFn_sqrts  1..rcJ|j|jdtdSNz ** Powr%rs rrjzPythonPrinter._print_FloatPow~~diiE1BCCrcJ|j|jdtdSrr%rs rrgz!PythonPrinter._print_PowByNaturalrrct|jdk7r tdd|j|jddS)Nr|"floor expects exactly one argument math.floor(rrr~rs r _print_floorzPythonPrinter._print_floors? tyy>Q  !EF FT[[167q99rct|jdk7r tdd|j|jddS)Nr|'FloorToInt expects exactly one argumentrrrr~rs r_print_FloorToIntzPythonPrinter._print_FloorToInts? tyy>Q  !JK KT[[167q99rct|jdk7r tdd|j|jddS)Nr|'TruncToInt expects exactly one argumentz math.trunc(rrr~rs rrlzPythonPrinter._print_TruncToIntA tyy>Q  !JK KT[[167q99rct|jdk7r tdd|j|jddS)Nr|$ceiling expects exactly one argument math.ceil(rrr~rs r_print_ceilingzPythonPrinter._print_ceilings? tyy>Q  !GH HDKK ! 56a88rct|jdk7r tdd|j|jddS)Nr|&CeilToInt expects exactly one argumentrrrr~rs r_print_CeilToIntzPythonPrinter._print_CeilToInts? tyy>Q  !IJ JDKK ! 56a88rct|jdk7r tdd|j|jddS)Nr| Abs expects exactly one argumentzabs(rrr~rs r _print_AbszPythonPrinter._print_Abss? tyy>Q  !CD Ddkk$))A,/022rct|jdkr tdddjt |j |jdS)Nz"Max expects at least two argumentszmax(, rrrrNjoinmaprrs r _print_MaxzPythonPrinter._print_MaxF tyy>A  !EF FdiiDKK ;<=Q??rct|jdkr tdddjt |j |jdS)Nrz"Min expects at least two argumentszmin(rrrrs r _print_MinzPythonPrinter._print_Min rrct|jdk7r tdd|j|jddS)Nr| cos expects exactly one argumentz math.cos(rrr~rs r_print_OpaqueUnaryFn_cosz&PythonPrinter._print_OpaqueUnaryFn_cos? tyy>Q  !CD D4;;tyy|45Q77rct|jdk7r tdd|j|jddS)Nr|!cosh expects exactly one argumentz math.cosh(rrr~rs r_print_OpaqueUnaryFn_coshz'PythonPrinter._print_OpaqueUnaryFn_cosh? tyy>Q  !DE EDKK ! 56a88rct|jdk7r tdd|j|jddS)Nr|!acos expects exactly one argumentz math.acos(rrr~rs r_print_OpaqueUnaryFn_acosz'PythonPrinter._print_OpaqueUnaryFn_acosrrct|jdk7r tdd|j|jddSNr|z sin expects exactly one argumentz math.sin(rrr~rs r_print_OpaqueUnaryFn_sinz&PythonPrinter._print_OpaqueUnaryFn_sin$rrct|jdk7r tdd|j|jddS)Nr|!sinh expects exactly one argumentz math.sinh(rrr~rs r_print_OpaqueUnaryFn_sinhz'PythonPrinter._print_OpaqueUnaryFn_sinh*rrct|jdk7r tdd|j|jddS)Nr|!asin expects exactly one argumentz math.asin(rrr~rs r_print_OpaqueUnaryFn_asinz'PythonPrinter._print_OpaqueUnaryFn_asin0rrct|jdk7r tdd|j|jddS)Nr| tan expects exactly one argumentz math.tan(rrr~rs r_print_OpaqueUnaryFn_tanz&PythonPrinter._print_OpaqueUnaryFn_tan6rrct|jdk7r tdd|j|jddS)Nr|!tanh expects exactly one argumentz math.tanh(rrr~rs r_print_OpaqueUnaryFn_tanhz'PythonPrinter._print_OpaqueUnaryFn_tanh<rrct|jdk7r tdd|j|jddS)Nr|!atan expects exactly one argumentz math.atan(rrr~rs r_print_OpaqueUnaryFn_atanz'PythonPrinter._print_OpaqueUnaryFn_atanBrrct|jdk7r tdd|j|jddS)Nr|z!log2 expects exactly one argumentz math.log2(rrr~rs r_print_OpaqueUnaryFn_log2z'PythonPrinter._print_OpaqueUnaryFn_log2Hrrct|jdk7r tdd|j|jddS)Nr|'RoundToInt expects exactly one argumentround(rrr~rs rrnzPythonPrinter._print_RoundToIntNs? tyy>Q  !JK K DIIaL12!44rct|jdk7r td|j\}}t|tj s t dd|j|d|dS)Nr*RoundDecimal expects exactly two argumentsz,ndigits must be an instance of sympy.Integerrrr)rrrN isinstancerGInteger TypeErrorr)rr numberndigitss rrpz!PythonPrinter._print_RoundDecimalTsd tyy>Q  !MN N))'5==1JK K F+,Bwiq99rcd}t|jD]@\}}|j|}|dk(r|}|j|}||}5d|d|d|d}B|r|SdS)NTr7z if z else r0)reversedrrrr resultexpr_icond_iexpr_strcond_strs r_print_PiecewisezPythonPrinter._print_Piecewise]s"&tyy1 INFF{{6*H~! ;;v.>%F  $xjvhaHF I v(S(rN))rsrtrurGrwrFrWrrrrYr\r^rar<rcrrrjrgrrrlrrrrrrrrrrrrrrrrnrprrxrrrzrzs??5::?#?&DuzzDcDCejjCSC 5:: # EJJ3EJJ3uzzcJUZZJCJEJJ3 JejjJSJ 7 7 7/ejj/S/DEJJD3DD DsD::: :ejj:S: :ejj:S:95::9#9 9UZZ9C9 3uzz3c3@uzz@c@ @uzz@c@ 8UZZ8C8 9ejj9S9 9ejj9S9 8UZZ8C8 9ejj9S9 9ejj9S9 8UZZ8C8 9ejj9S9 9ejj9S9 9ejj9S9 5ejj5S5 : :s:)UZZ)C)rrzceZdZdejdefdZdejdefdZdejdefdZdejdefdZ dejdefdZ dejdefdZ dejdefd Z dejdefd Z dejdefd Zdejdefd Zdejdefd ZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZdejdefdZ dejdefdZ!dejdefdZ"dejdefd Z#dejdefd!Z$dejdefd"Z%dejdefd#Z&dejdefd$Z'dejdefd%Z(dejdefd&Z)dejdefd'Z*dejdefd(Z+dejdefd)Z,dejdefd*Z-y+), CppPrinterr r ctjdvrdnd}t|}|tkDs |tkrt |dt |tk(r|dk7r tdd|dS||S) N)darwinwin32LLLz too big to convert to rz#unexpected minimum index type valuez(-1z << 63))sysplatformrMINDEX_TYPE_MAXINDEX_TYPE_MIN OverflowError INDEX_TYPErN)rr suffixis r_print_IntegerzCppPrinter._print_Integerss~)<<# I ~ ^!31#%.s- ADD  c:f#5#; < r ?  : r)rr cpqs` r _print_WherezCppPrinter._print_Wheres5 HL  1aCs#aS!!rcN|j|jdt|S)Nz || rrs rrzCppPrinter._print_Orrrcd}t|jD]T\}}|j|tddz }|dk(r|}*|j|tddz }||}K|d|d|}V|rd|dSdS) Nr1r2Tr r r7rr)rrrrrs rrzCppPrinter._print_Piecewises"&tyy1 CNFF((F1Cc1IJH~!,,VZ5G#5MN>%F (zXJc&BF C!'6(!}/C/rc |j\}}}|j|}|dk7r0|j|}|jr d|d|d}n d|d|d}|j|}dtd|d td|d S) Nr|,c10::div_floor_integer(static_cast(), static_cast()),c10::div_floor_floating(static_cast(), static_cast(z (static_cast<>(z) % static_cast<)rdoprint is_integerrrs rrz!CppPrinter._print_ModularIndexingsii 3 LLO !8,,s#CB1#E]^a]bbdeB1#E\]`\aacdll3zl"QC/? |2cURTUUrc|j\}}|j|}|j|}|jr d|d|dSd|d|dS)Nrrrrr)rrrrs rr<zCppPrinter._print_FloorDivsc3 LLOll3 ??A!D\]`\aacd d=aS@WX[W\\^__rct|jdk7r tdd|j|jdd}|jr dt d|dS|S)Nr|r std::floor(rr static_castQ  !EF F$++diil34A 648OOj\A3a0JJrct|jdk7r tdd|j|jdd}|jr dt d|dS|S)Nr|rrrrrrr r!s rrzCppPrinter._print_FloorToInts^ tyy>Q  !JK K$++diil34A 648OOj\A3a0JJrct|jdk7r tdd|j|jdd}dtd|dS)Nr|r std::trunc(rrrr)rrrNrrr!s rrlzCppPrinter._print_TruncToIntsS tyy>Q  !JK K$++diil34A 6j\A3a00rct|jdk7r tdd|j|jddS)Nr|z)TruncToFloat expects exactly one argumentr%rrr~rs rrrzCppPrinter._print_TruncToFloats? tyy>Q  !LM MT[[167q99rct|jdk7r tdd|j|jddS)Nr|r}static_cast(rrr~rs rrWzCppPrinter._print_ToFloats? tyy>Q  !GH H%dkk$))A,&?%@BBrcv|j\}}|j|}|j|}d|d|dS)Nz c10::div_mod(rr)rrrs rrazCppPrinter._print_PythonMods?3 LLOll3qcC5**rcn|j\}}d|j|d|j|dS)Nr(z) / static_cast(rrr)rr lhsrhss rrczCppPrinter._print_IntTrueDivs@99S&dkk#&6%77OPTP[P[\_P`Oaabccrc|j\}}|dk(rd|j|dStdt|)Nrz(1 << (rrf)rrrUrVrOs rrgzCppPrinter._print_PowByNaturalsJII c 19T[[-.b1 1!6tDzl C  rcn|j\}}d|j|d|j|dS)N std::pow(rrr+rOs rrjzCppPrinter._print_FloatPows8II c4;;t,-R C0@/ACCrc|j\}}|dk(s|dk(r"|j|}|dk(rd|dSd|dS|jrt|}|dkDr|j |g|zdt d}nO|d kr.d |j |gt |zdt dzdz}n|d k(rd |j|z}nd }|jr d td|dS|Sd|dt|dS)Nr2g std::sqrt(rz1.0/std::sqrt(rrrKz1.0/(z1.0/z1.0rrr0r) rrrrMrrabsrfloat)rr rPrQr"s rrRzCppPrinter._print_PowsII c #:;;t$D+.#:ZvQ' S^D6QR;S S >>c(CQwNND6C<j6GHrnndVc#h%6Z=NOP T[[..8<\*Rs!4 NQ NtfBuSzl!4 4rc|jdk(r|j}n|jd|jd}|jr dtd|dS|S)Nr|z.0/z.0rrr)rrrrr!s r_print_RationalzCppPrinter._print_Rational sR 66Q;66(A66(#dffXR(A48OOj\A3a0JJrct|jdk7r tdd|j|jdd}|jr dt d|dS|S)Nr|r std::ceil(rrrrr r!s rrzCppPrinter._print_ceilings^ tyy>Q  !GH HTYYq\231 548OOj\A3a0JJrct|jdk7r tdd|j|jdd}|jr dt d|dS|S)Nr|rr9rrrrr r!s rrzCppPrinter._print_CeilToInts^ tyy>Q  !IJ JTYYq\231 548OOj\A3a0JJrc |jDcgc]}|j|}}t|dk(rdtd|ddtd|dd Sdd j |zd z}d td|d Scc}w) Nrzstd::min(static_cast*:,ba Q[P\\^_cde_f^ggij jtyy&,Bzl"RD2 2 3A8c |jDcgc]}|j|}}t|dk(rdtd|ddtd|dd Sdd j |zd z}d td|d Scc}w) Nrzstd::max(static_castz std::max(std::nearbyint(1ez * z) * 1er)rrrNr ValueErrorrr)rr rr number_strs rrpzCppPrinter._print_RoundDecimals tyy>Q  !MN N))   !|$%RSSabiajjkl &&vz%/@A 6wis:,fV]U]T^^_``rcy)Ntruerxrs r_print_BooleanTruezCppPrinter._print_BooleanTruesrcy)Nfalserxrs r_print_BooleanFalsezCppPrinter._print_BooleanFalsesrcy)Nz'std::numeric_limits::infinity()rxrs rrYzCppPrinter._print_Infinitys8rc*d|j|S)N-)rYrs rr\z"CppPrinter._print_NegativeInfinitys4''-.//rcy)Nz(std::numeric_limits::quiet_NaN()rxrs rr^zCppPrinter._print_NaNs9rN).rsrtrurGrwrFrrrrrr<rrrlrrrWrarcrgrjrRr7rrrrrrrrrrrrrrrrrnrprZr]rYr\r^rxrrrrrs 5:: # """ CejjCSC0UZZ0C0$ V5:: V# V`EJJ`3`KKKKejjKSK1ejj1S1: :s: C5::C#C +UZZ+C+ dejjdSd   s DEJJD3D 5uzz5c5:KEJJK3KK5::K#KKUZZKCK3uzz3c33uzz3c38uzz8c8 8UZZ8C8 9ejj9S9 9ejj9S9 8UZZ8C8 9ejj9S9 9ejj9S9 8UZZ8C8 9ejj9S9 9ejj9S9 9ejj9S99ejj9S9:ejj:S: a as auzzc s9EJJ9390EJJ030:uzz:c:rr) rrGsympy.printing.precedencerrsympy.printing.strrrrrr rzrrxrrrdsN <)  | *| ~a)Ka)Hh:h:r