ۓ,jNSr/SQrSSKrSSKrSSKrSSKrSSKJr SSK r SSK r SSK r SSK rSSKJr SSKJr SSKJrJrJrJr SSKrSSK Jr SS KJr S rS rS r\ R<"5rS r \ R<"5r!S r"\ R<"5r#Sr$"SS\%5r&"SS\%5r'\("\'RRRU55\'l+"SS\'5r,Sr-"SS\'5r."SS\'5r/"SS\'5r0SSK17 SSK1Jr2 \\2- rSSK37 SSK3J4r4 SSK3Jr5 \\5- rSSK6J7r7 SS K6J8r9 \S!S"/- r"S#S$\'5r:S%r;\_props_for..(sU_`ackUlrr)_props_for_dictsetinspect getmembersclsps r _props_forr/&sK /!"'2D2DSJl2m#n2mQaD2m#no 3 $osAcU[;aP[R"US5Vs/sH%n[USRS5(dM#UPM' sn[U'[U$s snf)Nc"[U[5$r r!r$s rr& _dep_props_for...sR\]^`hRirr deps)_dep_props_for_dictr*r+hasattrfgetr,s r_dep_props_forr7,sl %%/6/A/A#Gi/j$J/j!nuvwxyvzvvBHoIA/j$JC s ##$Js "A$ A$c<U[;a[[RR[ UR 5R [ UR555[U'[U(a)[S[U5<S[[5<35eg)Nz In class z/, don't use reserved keywords in terms/dterms: ) _kw_conflict_dictr _reserved_kw intersectionr)termsuniondterms Exceptionstr kw_conflict)r-s r_check_kw_conflictrB3sp ##!#!=!=c#))n>R>RSVWZWaWaSb>c!d#Z]^aZbdghsdtvwwrc"\rSrSrSrSSjrSrg)Term:rNcXlX lX0l[RUl[=RS- slgNr )defaultdescdrrDcreation_counter)selfrHrIrJs r__init__ Term.__init__<s2  !% 5 5 "r)rKrHrIrJ)NNT)__name__ __module__ __qualname____firstlineno__rKrM__static_attributes__rrrDrD:s #rrDc4^\rSrSr/rS/rSr0r0rSSS.r Sr Sr Sr U4Sjr \S 5r\S 5r\S 5r\S 5rS rSrSr\S5r\S5r\S5r\S5rS`SjrSaSjrSrSrSr\S5r \S5r!Sr"Sr#Sr$Sr%Sr&S r'S!r(SbS"jr)S#r*SbS$jr+SbS%jr,S&r-S'r.S(r/S)r0S*r1SbS+jr2S,r3S-r4S.r5S/r6S0r7\S15r8S2r9SbS3jr:SbS4jr;SbS5jrSbS8jr?S9r@S:rAS;rBS<rCScS=jrDS>rE\S?5rFS@rGSbSAjrHSBrI\JR4SCjrLSdSDjrMSErNSFrOSGrPSbSHjrQSbSIjrRSbSJjrSSKrTSLrUSMrVSNrWSOrXSPrYSQrZSRr[SSr\STr]SUr^SVr_SWr`SXraSYrbSZrcS[rdS\reS]rfS^rgS_rhU=ri$)erGr%g@F) make_dense make_sparse undefinedNc Z>[U5S:a"[US5[S5:Xa[n[[U]U5nUR 5 [RUS[55 [RUSS5 [RUS[R"55 [RUSS[R"5S.5 [(a.[RUS05 [RUS S 5 [URR55H7upE[RUS U-UR!XE55 XB;dM5X$ M9 [#U5n0nUH)nUSn US R$R&n U SS S.Xy'M+ [RUSU5 U[:wa&[)U5HupXUR*U 'M O:[U5S:a+[,R."US[,R05US'UR35Vs0sH1oR4cMUR6[9UR45_M3 n nU R;U5 UR"S0U D6 U$s snf)Nrcg)NrrTrTrrr&Ch.__new__..XsAr _dirty_vars_itr_parents_cacherdrs _cache_info_statusnewz_%sr T)r3value out_of_date_depends_on_depsr%rT)lentypersuperr__new__ setup_termsobject __setattr__r)weakrefWeakKeyDictionaryrlist_default_kwargsitemsgetr7r6r3 enumerate term_ordernpasarrayfloat64parm_declarationsrHnamerupdate)r-argskwargsresultr} default_valpropscpdr. func_namer3idxadefs __class__s rrm Ch.__new__Vs  t9q=T$q']d:.>>Cr3', 6=#%866406:w/H/H/JK684@Y@Y@[-\] 5   v}b 9   vy% 8!%c&9&9&?&?&A!B D   vut|VZZ5R S~L"Cs#A!IQ499>>D&*T$OCN 6#5s; "9#D/./s~~c*+* Yq[**T!Wbjj9F3K696K6K6Mg6MQZQZ,!)),,6Mg F T hs J($J(c.URUR;ac[R"5n[R "US5n[ USS9Hup4X4lXAU'M XRUR'URUR$)Nc"[U[5$r )r"rDr$s rr&&Ch.parm_declarations..s :a;Nrc USR$rG)rKr$s rr&rsqt?T?Tr)key)rO _cached_parms collections OrderedDictr*r+sortedr})r-parameter_declarations parametersr}decls rr|Ch.parm_declarationss < " ++C1NOJ$Z5TU  /3t,V/E  cll +  ..rcD[U5UR;agU[:XagUR5nUR[RLa/UlO1[ UR[ 5(aUR4UlUR[RLa/UlO1[ UR[ 5(aUR4Ul[UR5[UR5-n[U5S:aUS:XdeUVs/sHo3RPM snUl UVs/sH!o3R(dMURPM# snUlUVs/sH!o3R(aMURPM# snUlOA[US5(d0[UR5[UR5-Ul [U5 SUR[U5'gs snfs snfs snf)NrrxT)id _setup_termsrr|r>r"r@r<rjr}rxrJr5rsrB)r-r|len_oldstyle_parmsts rrnCh.setup_termss c7c&& & "9 113 :: "CJ  C ( (**CJ 99 CI  3 ' ' CI!_S^;  !A %%q( )(.?@.?ff.?@CN*;D*;Qtt&!&&*;DCJ):G):A$$):GCI3 --!%cii4 3C!C3$(C!ADGs(H H!H=HHch[US5(a UR$URR$)Nlabel)r5rrrOrLs r short_name Ch.short_names'$T733tzzP9P9PPrc $[UR5[UR5-nUR5 UR[ UVs/sH.o"UR ;dMU[UR U54PM0 sn54$s snf)z Semantic id.)rsr<r>sortrtuple__dict__r)rLpnamesks rsidCh.sidsqdjj!$t{{"33 &&g&QY]YfYfTf'@2dmmA.>+?'@&&g hii&gs B 'B cB[U[U5S:aUS9$USS9$)Nr rrnewshape)reshaperjrLrs rr Ch.reshapes%D ! IIaIIrc[USS9$)Nr)rrs rravelCh.ravels--rc[U5$r )rrs r__hash__ Ch.__hash__s $xrc.URR$r )rbndimrs rrCh.ndimvv{{rc.URR$r )rbflatrs rrCh.flatrrc.URR$r )rbdtypers rrCh.dtypevv||rc.URR$r )rbitemsizers rr Ch.itemsizesvvrcUS:Xa0nURnSn[UR5H{nURUn[ US5(dM%UR nXq;aXaU'O'URUXLaXURU'SnUR USS9(dMySnM} U(dgU(aUR USS9 gg)NFr>T)iterate)rbrsr>rr5rremove_redundancy)rLcacher_replacement_occurredpropnameproprs rrCh.remove_redundancys D=EA$T[[)H==*D4**((C!c x( :*/* h''+$%%eU%;;'+$*$&&ud&;rcUc[Rn[US5(aPURS[ U5-S-UR [ R"URS-54-5 URH1n[X5n[US5(dM!URSUS9 M3 U(aFURS[ U5-S -[ R"URS-54-5 gg) Nrz%s: %.ze | r>F) print_newlinewhere_to_printz%.ze ) rstderrr5rr@rrysumrbr>getattrprint_labeled_residuals)rLr num_decimalsrdtermdts rrCh.print_labeled_residualss  ! ZZN 4 ! !  (S->">"GDJJXZX^X^_c_e_egh_hXiKj!j k[[E%Br8$$**~*^!   $\):":U"BrvvdffVWiGXFZ!Z [ rcUR$)@Default method for objects that just contain a number or ndarrayr$rs r compute_r Ch.compute_rs vv rcXLa?[R"URRURR5$g)rN)speyer%sizerLwrts rcompute_dr_wrtCh.compute_dr_wrts+ ;66$&&++tvv{{3 3rcUR5 URU5nUbU$Un[UR[5(aGUR nX0LagURU5nUbO![UR[5(aMGUcgUR5 URU5R nURX$5$r )_call_on_changedr issubclassrPermuterT _superdot)rLrrinnerjacs r_compute_dr_wrt_slicedCh._compute_dr_wrt_sliceds  $$S)  M EU__g66=,,U3%U__g66~  "$$U+--C>>&. .rc.URR$r )rbshapers rrCh.shape;rrcB[R"UR5$r )ryprodrrs rrCh.size?swwtzz""rc,[UR5$r )rjrbrs r__len__ Ch.__len__D466{rc:SSKJn UR"U/UQ70UD6$)Nr ) optimization)rminimize)rLrrrs rr Ch.minimizeGs "$$T;D;F;;rcUR$r )rbrs r __array__ Ch.__array__Ks vv rc|[[[UR5U/-55Ul[XU5 gr )rsr)r>setattr)rL dterm_namers r add_dterm Ch.add_dtermQs-3tDKK0J<?@A %(rc[U5$r )copyrs rr Ch.copyU DzrcR[R"UR5nUS US U$)Nr_r`) external_copyr rrLrs r __getstate__Ch.__getstate__Xs+##DMM2 :  8  rcb[R"5US'S[R"5S.US'[RUSU5 [ UR 5R [ URR555Hn[XURU5 M g)Nr_rar`r) rqrrrorpr)r>r;rkeysr)rLdrs r __setstate__Ch.__setstate___s113*  )B)B)DE( 4Q/T[[!..s4==3E3E3G/HIA DT]]1- .Jrc[U5[:XaIUS:Xa,URRU5 UR U5 [ R XU5 gXR;nXR;nU[UR5;nU(aU(d[U5[:wa[US5(d [U5n[X5(ag[X5nXR;aMURUSRU5 [URUS5S:Xa URU XR;aS[!U/50URU'O!URUSRU5 U(dU(a>URRU5 UR#U/5 UR U5 [ R XU5 g)Nr%r>varnamesr)rkrr]add clear_cacherorpr>r<r/rr5rr_removerjr)_invalidate_cacheprop_names)rLr}rgitrname_in_dterms name_in_terms name_in_propsterms rrpCh.__setattr__is : s{  $$T*  %   t5 1 , *  4>> :: -DJ"4D5(++5 t""t*==(MM$' 3::4@4==.z:;q@ MM$/>>)(2CK'@t$t$Z044T: ]     &  , ,dV 4   S !4u-rc[U5n[URR55H-up4[ UR US55S:dM(SUS'M/ g)Nr3rTrh)r)rsrirurjr;)rLnamesnamesetrvs rrCh._invalidate_cacheprop_namessRe* !6!6!?LI7''& 23a7#'- @rc zU/n[/5nSn[U5S:GaUS- nUR5nUbXR:XaM9[ U5U;aSUR S'UR SR 5 Xl[URR55H]upg[RUSURRUS55 URUS5 URU5 M_ UR![ U55 [U5S:aGMU$)Nrr rbrcr]r)r)rjpopr^rr`clearrsr_rurorpr]r=rappendr)rLrtododone nodes_visitednextparent parent_dicts rrCh.clear_caches v2w $i!m Q M88:D3 >$xt##' C  E"((* +/ 0C0C0E+F'F&&v}f>P>P>V>VWbcmWn>op66{:7NOKK',GD"$i!mrc dXRS;aSURSU'[US5(aXR;aSURU'UbX R:wa[ UR R 55Hup4XRS;d [US5(aXR;aURXS9 [RUSURRUS55 URUS5 M [RUSU5 g)Nrc dr_cached)rrr]rr^) r`r5r3r^rsr_ruclear_cache_wrtrorpr]r=r)rLrrr/r0s rr4Ch.clear_cache_wrts ++e$ $&*DKK s # 4 % %#*?"&DNN3  ;#*'+DMM,?,?,A'B#--..76;3O3OTW[k[kTk**s*<""6=&:L:L:R:RS^_iSj:kl22;z3JK (C 4-rc[US5[US5:wa [S5e[UR5[UR5-Vs/sHn[X5(dMUPM snHDn[ X5nXQLa[ XU5 M [US5(dM3URX5 MF U$s snf)Nr>zGEither "old" and "new" must both be "Ch", or they must both be neither.)r5r?rsr>r<rrreplace)rLoldrfr term_namer s rr7 Ch.replaces C "gc8&< <ef f%)$++%6tDJJ7G%G\%G7SWK[!%G\I4+D{-x(( S& ]  ]s B>+B>c j[UR5n[UR55nUR U5nUR U5n[ U5[ U5-n[U5S:a=USSHnURXqUS5 M URUSXSS5 gg)Nrri') r/rr)rr; differencersrjrp)rLrr kwarg_keyskwsecondkwfirstkwallrs rr)Ch.sets4>>*' **51''1W X. u:>3BZ  1It4   U2YRy(94 @ rc[U5[:XaXL$URVs/sH n[X5(dM[ X5PM" nnX;=(d* [ UVs/sHoDR U5PM sn5$s snfs snfr )rkrr>r5rany is_dr_wrt)rLrrdterms_we_havers rrD Ch.is_dr_wrtsp : ; <@KK`K57SWK_.'$.K`$W~,V~![[-=~,V(WWa,VsBB"Bc&UR[L$r )rrrs ris_ch_baseclassCh.is_ch_baseclasss~~##rcURn[R"[R"U55R U5R U5nUR 5nURn[XUS9$)N)ridxspreferred_shape)rryarangerr __getitem__rSelect)rLrrtmprKrs rrNCh.__getitem__sX ii'//6BB3Gyy{99BBrcH[US5(a [S5e[U5[:XaB[R "UR 5nURX5 URSXCS9 gUnUR5(dN[UR[5(a URnO [S5eUR5(dMNXnURU5nURnUR nUR#U[R$"U5R'55n UR#XpR#XeR R'555n [(R*"U 5(a$[R,"U R/55n [(R*"U 5(a$[R,"U R/55n UR U R1U5-U R1U5- Ulg)Nr>z1Can't assign a Ch objects as a subset of another.r%)rz+Can't set array that is function of arrays.)r5r?rkrry atleast_1dr% __setitem__rprHrrrrdr_wrtrrrrzrrissparsearraytodenser) rLrrgrdatarrJdr_rev inner_shapet1t2s rrTCh.__setitem__s} 5( # #OP P : ==(D   S (   S$  0 E++--eoow77!GGE#$QRR ++-- 9DU#BTTF++K 5(9(?(?(ABBr77==?(KLB{{2RXXbjjl%;{{2RXXbjjl%;gg ; 77"**[:QQEGrc,[UR5$r )r@rbrs r__str__ Ch.__str__%rrc^[RU5S-[UR5-$r)ro__repr__r@rbrs rrc Ch.__repr__(s#t$t+c$&&k99rc6URR5$r )rb __float__rs rrf Ch.__float__+svv!!rc6URR5$r )rb__int__rs rri Ch.__int__.svv~~rcgr rT)rLr<s r on_changed Ch.on_changed1s rc[U5$r  transposers rrCh.T4s rc[U/UQ76$r ro)rLaxess rrp Ch.transpose8s%%%rc[X5$r )squeezerLaxiss rrv Ch.squeeze;s t""rc[XS9$N)rx)meanrws rr|Ch.mean>s D$$rc[XS9$r{)rrws rrCh.sumAs 4##rc>[US5(a UR5upU(dU5e[US5(aSUl[UR5S:a;UR UR5 [ RUS[55 gg)Nis_validrerfrr]) r5rrerjr]rlrorpr))rLvalidityrs rrCh._call_on_changedDs{ 4 $ $ MMOMH S 8 4 # # DL t 1 $ OOD,, -   t]CE : %rcUR5 URSc[R"[R"UR 55[R SS9URS'URSR5URS'SURSRl URS$)NrbC)rorderrviewF) rr`ryrzrSrr{viewflags writeablers rrbCh.rPs  ;;s  #!zz"--8H*IQSQ[Q[cfgDKK #';;s#3#8#8#:DKK 38DKK & & 0{{7##rc&^^TcgTcg[T[R5(a#TRS:XaTR 5Sm[T[R5(a#TRS:XaTR 5Sm[T[ R 5(d[T[ R 5(aTT-$[T[5(a-[TRSTRS4UU4Sj5$[T[5(aY[R"T5(a-[TRSTRS4UU4Sj5$TRT5$[5n[TT5ummU"5S:a[SRU"555 [R"T5(dJ[R"T5(a/TR RTR 5R $TRT5$!["aJnSSKnSSKnUR)UR*S9 [,(aSSKoR15 SnAgeSnAff=f)Nr rcD>TRTRU55$r dotr%lhsrhss rr&Ch._superdot..lsswwWZW^W^_`WaObrcD>TRTRU55$r rrs rr&rpsSVSZSZ[^[b[bcd[eSfr皙?z.convert_inputs_to_sparse_if_necessary in {}sec)file)r"ryndarrayrrnumbersNumberr rrrVrr rrformatrr?r traceback print_excrrpdb post_mortem) rLrrprofilertm_maybe_sparseerrrs `` rr Ch._superdotZs) {{#rzz**sxx{iik!n#rzz**sxx{iik!n#w~~..*S'..2Q2QSy #~..%syy|SYYq\&BDbcc#~..;;s##)399Q<1*FHfgg773<'$gOzBROKEN!zBroken Should be Ch objectrr c X-$r rTr%ys rr&Ch.lmult_wrt..r)rrr*rr)r>rr5rDr"rprintr? lmult_wrtrjr) rLrrrc direct_drrr. indirect_drrs rr Ch.lmult_wrts+ ; //4   JJt~~c5 6T[[!A a Aq(## S9I9I!!R(()$#$@AAkk$..>Y>YZ[>\*]_bc *JJ{+" s8Q;F Xq[VF  ,c2F rcURU5nUcg[U[5(dURX#5$UR U5$r )rr"r rmatmat)rLrrrJs r compute_lopCh.compute_lopsF  ( ( - :d.8n.M.Mt~~c&aSVS]S]^`SaarcvUR5 /nURX5nUbURU5 [UR5Hdn[ X5n[ US5(dM!XaLdM'URXb5nUcM=URX5nUcMSURU5 Mf [[U55H6n[R"X55(dM"X5R5X5'M8 [U5S:XaSn U $[U5S:XaUSn U $[SU5n U $)Nr>rr c X-$r rTrs rr&Ch.lop..rr)rrr*r)r>rr5loprangerjrrVrXr) rLrrrcrrr. lhs_for_childrrs rrCh.lops $$S.  JJy !T[[!A Aq(## $ 0 0 8  ,"#%%";K". ;/"s3xA{{36"")! s8Q;F Xq[VF  ,c2F rcNURU5nUcgURX25$r )rr)rLrrrJs r compute_ropCh.compute_rops(  ( ( - :d~~b&&rc v ^[5nSUlUR5 /mXRS;aY[(a<XR ;aSUR U'UR U==S- ss'SUlURSU$URU5nUbTRU5 [(aSUl[[UR55n[UR5RUR[URR!5555GH n[#X5n[%US5(dM"XLdM(Sn U(aURU5n ['U [(5(aKUR+5 UR-U5n UR/5 U S:waU R1U 5OSn OSUR3X5n OAUR+5 UR-XS9n UR/5 U bUR5XS 9n U cMTRU 5 GM [7T5S:XaSn Oz[7T5S:XaTSn Oe[8R:"TV s/sHn ['U [(5PM sn 5(d[=S T5n O[)TSR>U4S j5n U b[@RB"U 5(d[5n[8RD"U 5nU"5S :a[GS RIU"555 US:Xd-[%U S5(aoU RJ[MU5- S:aS[5n[@RN"U 5n SSK(nURS5 [GSRIU"555 U bF[@RB"U 5(d+['U [(5(d[8RT"U 5n [7[WURXR!555S:waXRSU'[(aSUl[#USS5(a+[@RB"U 5(aU R[5n [#USS5(a1[@RB"U 5(d[@RN"U 5n U"5S :aJ[GSRIUR\UR\U"5[@RB"U 555 U $s sn f)NTrcrr cachedpendingr>)r)rc X-$r rTrs rr&Ch.dr_wrt..sQSrc&>^[U4SjT5$)NcJ>URT5URT5-$r r)rbr%s rr&-Ch.dr_wrt....s$UVUZUZ[\U]^_^c^cde^fUfrr)r%rcs`rr&rsHfgjAkrrzcount_nonzero in {}secrg$@z$converting result to sparse in {}secr, _make_denseF _make_sparsez$dx of {} wrt {} in {}sec, sparse: {})/r called_dr_wrtrr`rrdrerr*r)r/rr>r;r=rrrr5r"r pauserUresumerrrrjryrCrrrrV count_nonzerorrrfloat csc_matrixgccollect atleast_2drsr_rXr)rLr reverse_moder tm_dr_wrtr propnamesrr.rrdr2rr tm_nonzerononzerotm_convert_to_sparserrcs @rrU Ch.dr_wrtsG !  ++e$ $u...,-D$$S)  %)%' ;;u%c* *//4  JJy ! 5$DL 4>>23 T[[!..ys4==CUCUCW?X/YZA Aq(## " 55a8C!#~66!)hhsm!((*9<cjjo$ &'kk#&; OO%((3(:C$$&&*&6&6q&6&B *JJ{+3[6 s8Q;F Xq[VF66#F#Q:a8#FGG 0#6'A 6kl  bkk&&9&9J&&v.G|c!,33JLAB!|wvv666;;w;W[_;_',w$v. :AABVBXYZ  V)<)UR"S0UD6 UR$)NrT)r)rb)rLrs r__call__ Ch.__call__Es 6vv rcS$)z Used as fn in loop_children_do c[USS5$)NrF)rr$s rr&Ch.reset_flag..Rs OU;rrTrs r reset_flag Ch.reset_flagMs <;rcU"U5 URHhn[X5(dM[X5n[US5(d[US5(dMD[US5(dMWURU5 Mj g)Nr>r<loop_children_do)r>r5rr)rLfnrdtvals rrCh.loop_children_doTsa 4[[Et##,5(++wug/F/Fu&899..r2 !rc ^^ SSKnSSKnSSKn[(dS5eURR S5nUU 4Sjm SSR [[T "US5555-nURR Xa5nURR XaS -5n [US 5n [U S 5n U RU5 U R5 URS S S 9n URS SSU RU R/5 SSS5 SSS5 SSKn U RSW R-5 UR!UR"5 g!,(df  NT=f!,(df  N]=f)z Show tree and cache info with color represent _status Optionally accpet current_node arg to highlight the current node we are in rN8Please use dr tree visualization functions in debug modeprofilesc >SSSSS.n[US5(a URnU<S[URR5<S3n/n[US 5(dU$UR GHn[X5(dM[ X5n[US 5(d[US 5(dME[US5(a [ US5OUnU<S[URR5<S3nS [U5-nS [U5-nS n [US 5(aX%ROSn UT:XaSn [U[5(a[UR5S:aSn URR5HupU cM [R"U 5n U R nU (a5U R"SU R"S-n[U R$5nO[&R("U 5nU SXU4-- n M O[UR*S5S:aSn UR*SR5HupU cM [R"U 5n U R nU (a5U R"SU R"S-n[U R$5nO[&R("U 5nU SXU4-- n [US5(dMU SUR,U -- n M X7<SU<S3/- nUU<SU<SU <SU <S3/- nUT"[ X5U5- nGM U$)Ngreyredyellowgreen)rfrrr,r ()r>r<aaa%drrebluerz dr_cached r z! sparse: %s size: %d nonzero: %d rcz_cache rdz cache hit:%s  -> ; [label="rz ", color=z, style=filled];)r5rr@rrOr>rrrer" Concatenaterjr3rurrVrrrYryrr`rd)rLmy_name color_mappingrrr child_labelsrcdstscolorrr%rVrr current_node string_fors rr&Ch.show_tree_cache..string_forjs%+uxY`aMtW%%**#*C0G0G,HIGF4** 4''#D0Euh//75'3J3JAHPWAXAXgeW&=^c 3>EOOD\D\@]&^ %D2%E3@Gy@Y@Y mm <_e L0$*E%e[99c%//>RUV>V -A(-(=(=(?#$=/1{{1~H+,66D'//0wwqzAGGAJ/F25aff+242B2B12E$%)PT\dkSl)l$lA)@!e!459 *A(- U(;(A(A(C#$=/1{{1~H+,66D'//0wwqzAGGAJ/F25aff+242B2B12E$%)PT\dkSl)l$lA'.um'D'D()-?%BSBSTUBV-V(V)D$#s#;"<<$'a$@#AA*WT-A5"IIq%rMrdigraph G { %s }rroot.pngwFsuffixdeleter-Tpng-ofile://)ostempfile subprocessrpathabspathjoinrsr)openrflushNamedTemporaryFilecallr} webbrowserrr)rLrrr r r  cache_pathdot_file_contents dot_file_name png_file_namedot_filepng_filerrs ` @rshow_tree_cacheCh.show_tree_cache^s3 uPPPuWW__Z0 B J1499T#jQUW]F^B_=`3aa Z7  Zv> - %mS)X01 #66fU6Sx}}hmm TU *&  HMM12 doo.*)& %s%5 E=AE,E=, E: 6E== F c ^SSKnSSKn[(dS5eU4SjmSSR[ [ T"USU5555-nUR 5nURU5 UR5 UR SSS 9nURS S S URUR/5 SSK nURS UR-5 g)Nrrc>[US5(a URnU<S[URR5<S3n/n[US5(dU$UR GH}n[X5(dM[ X5n[US5(d[US5(dME[US5(a [ US5OUnU<S[URR5<S3nS[U5-nS[U5-nX7<SU<S3/- nX7<S U<S 3/- nX%RS ;aURS Ubu[R"URS U5n URS URn [R"URS U5n US XXU 4-/- nO X8<S U<S 3/- nUT "[ X5XB5- nGM U$) Nrrrr>r<rrrr"];rcz4%s [label="%s is_sparse: %s size: %d nonzero: %d"];)r5rr@rrOr>rrr`rrVrryr) rLrrrrrrrrrVrrrs rr$Ch.show_tree_wrt..string_forstW%%**#*C0G0G,HIGF4** 4''#D0Euh//75'3J3JAHPWAXAXgeW&=^c 3>EOOD\D\@]&^ %D2%E3#s#;"<<g#F"GG,,u"55%,,u:Mc:R:^'){{5<<3Fs3K'LH#(<<#6s#;#@#@D&(&6&6u||E7J37O&PG"'`(+((/(1(1'22F#S+'N&OOF*WT-A5"NN'%(MrrrrrFrrrrr) r r rrrsr)rrrrr}rr) rLrr r rrrrrs @r show_tree_wrtCh.show_tree_wrtsuPPPu <1499T#jQUW]_bFcBd=e3ff..0()..fU.Kx}}hmmLM HMM12rc ^^SSKnSSKn[(dS5eUU4SjmSSR[ [ T"US5555-nT[ R:waSSKnURSXER5n[U5S:a[ R"[ R"UVs/sHn[U5PM sn55nUHxn [U 5U- n [U 5n S n U S :aU S -n S n U S :aU S -n S n U S :aU S -n Sn U S :aU S -n Sn UR!SU -SU S-SSU - S-X4-U5nMz UR#5n U R%U5 U R'5 UR#SSS9nUR)SSSUR*U R*/5 SSKnUR/SUR*-5 gs snf)z^Cachelim is in Mb. For any cached jacobians above cachelim, they are also added to the graph. rNrc<>[US5(a URnU<S[URR5<S3n/n[US5(dU$UR Hn[X5(dM[ X5n[US5(d[US5(dMD[US5(a [ US5OUnU<S[URR5<S3nS[U5-nS[U5-nX&<SU<S3/- nX&<S U<S 3/- nX'<S U<S 3/- nUT "[ X5U5- nM T [R:wa[US 5(aS UR;aS SK J n [URS R55HaupIS[U5-nS[U5-n[R "UR#U S55n U T S-S-:dMVUSXgU 4-/- nMc U$! S n N%=f)Nrrrr>r<rrrrrr`rcr)cPicklerz)%s -> %s [style=dotted,color="<<<%d>>>"];)r5rr@rrOr>rrryinfr` six.movesr$rsrur getsizeofdumps) rLrrrrrrrpicklerszcachelimrs rr Ch.show_tree..string_forstW%%**#*C0G0G,HIGF4** 4''#D0Euh//75'3J3JAHPWAXAXgeW&=^c 3>EOOD\D\@]&^ %D2%E3#s#;"<<g#F"GGk#J"KK*WT-A5"II%266!gdH&=&=%4;;BV7"&t{{5'9'?'?'A"BJE"RX.C!RY/C ]]6<<R+@A X_t34#NRU\^Q_#_"``#C&Ms &HHrrrz <<<(\d+)>>>rg@r%KMGrz<<<%s>>>z#%02x%02x%02x",label="%d%sr rFrrrrr)r r rrrsr)ryr&refindallDOTALLrjmaxrWintrsubrrrrr}rr)rLr,r r rr2strsrthe_maxrszpctr+unitrrrrs ` @r show_tree Ch.show_treesuPPPu) V1499T#jQUW]F^B_=`3aa rvv  ::n.?KD4y1}&&4*@4a3q64*@!ABA!!HW,EqBDEzd "Ezd "Ezd "Ezd "(*zA~?[_deh_hjknopunuwzmz|~_F@FHY)Z%#&..0()..fU.Kx}}hmmLM HMM125+As8G$ c## Uc [5nX;aU(a&[U[5(aURU5 UR U5 Uv [ US5(dSv UR H`n[ X5(dM[X5n[ US5(d[ US5(dMDURU5HnUv M Mb gg7f)zK Generator function that traverse the dr tree start from this node (self). Nr>r<) r)r"rsr*rr5r>r tree_iterator)rLvisitedr rchildnodes rr?Ch.tree_iterator@s ?eG   4.. D! KK J4**4''#D0Euh//75'3J3J$)$7$7$@D"&J%A % sBC +C<"Cc[U5$r )floorrs rrECh.floorXs T{rc[U5$r )ceilrs rrHCh.ceil[r rc[X5$r rrLothers rrCh.dot^s 4rc[XS9$N)rrx)cumsumrws rrP Ch.cumsumas ((rc[XS9$rO)aminrws rminCh.mind d&&rc[XS9$rO)amaxrws rr5Ch.maxgrVrcU$r rTrs r__pos__ Ch.__pos__msd{rc[U5$r )negativers r__neg__ Ch.__neg__ns htn,rc[XS9$N)rrrrKs r__add__ Ch.__add__ps cD&::rc[XS9$rbrcrKs r__radd__ Ch.__radd__qs cE&::rc[XS9$rbsubtractrKs r__sub__ Ch.__sub__s h&??rc[XS9$rbrjrKs r__rsub__ Ch.__rsub__t h&??rc[XS9$rbmultiplyrKs r__mul__ Ch.__mul__vrnrc[XS9$rbrtrKs r__rmul__ Ch.__rmul__wrrrc[XS9$N)x1x2dividerKs r__div__ Ch.__div__ys f&??rc[XS9$r|rrKs r __truediv__Ch.__truediv__zs &D*C#Crc[XS9$r|rrKs r__rdiv__ Ch.__rdiv__{s f&??rc[XS9$N)r%powpowerrKs r__pow__ Ch.__pow__}s ed&>>rc[XS9$rrrKs r__rpow__ Ch.__rpow__~s ee&>>rc$URU5$r )__and__rKs r__rand__ Ch.__rand__sdll5&99rc[U5$r )absrs r__abs__ Ch.__abs__s s4y(rc[X5$r )greaterrKs r__gt__ Ch.__gt__s GD$88rc[X5$r ) greater_equalrKs r__ge__ Ch.__ge__s M$$>>rc[X5$r )lessrKs r__lt__ Ch.__lt__s D$55rc[X5$r ) less_equalrKs r__le__ Ch.__le__s Jt$;;rc[X5$r ) not_equalrKs r__ne__ Ch.__ne__s Id$::r)rerr>)NT)TrNr )FN)NN)jrOrPrQrRr<r>__array_priority__rrrtrerrrm classmethodr|rnr#rrrrrrrrrrrrrrrrrrrrr rrrprrr4r7r)rDrHrNrTr`rcrfrirlrrprvr|rrrbrrrrrrUrrrrr ryr&r<r?rErHrrPrTr5r[r_rdrgrlrprvryrrrrrrrrrrrrrS __classcell__rs@rrrGs EUFML&+UCOGMH -^//))FQQjjJ.@ \$/@##< )/..`(,. A$X$C'RV:"  &#%$ ;$$+Z"Jb B' iX<< 3_/B+3Z"$Q3h'0 )'' #,::?????C?>>9(8>5;::rrc(\rSrSrSrSrSrSrSrg)rimtxveccURR[URRR 555R 5n[ URRR5S:a[URRRSS:a4URSURRRS45nU$)Nr r) rrr rrbrrjrrrs rrMatVecMult.compute_rsc$((**"2"2"456<<> txxzz 1 $)9)9!)rrrSrTrrrrs E F +rrc^[5mUHLn[U[5(aTRU5 M+UVs/sHnTRU5PM nMN U4SjnU$s snf)Nc>^^S[R"T5R;m[T5UU4Sj5nTUl[ U5nU$)Noutc>TRnURUnUSS:Xa#T(aUSUS'T"U/UQ70UD6US'SUS'US$)NrhTrgrF)rOri)rLrrrsdffuncwant_outs r with_caching5depends_on.._depends_on..with_cachingse I'' 2C=!T)$'LF5M#D:4:6:G %*M"w< r)r*getfullargspecrrr3r#)rrrrr3s` @r _depends_ondepends_on.._depends_onsMG2248=== t   ! ,' r)r)r"r@r) dependenciesdeprrr3s @rrrsX 5D c3   HHSM"% &#QTXXa[# &  $ ) 'sA&c$\rSrSrSrSrSrSrg)rir$cPURULdeURR$r )r%rbrs rrChHandle.compute_rs!vvT!"!vvxxrc"XRLaggrGr$rs rrChHandle.compute_dr_wrts &&= rrTNrOrPrQrRr>rrrSrTrrrrs FrrcX^\rSrSrSS/r/rSS/rSrS SjrU4Sjr Sr Sr S r U=r $) rilmb initial_argsc[U5R[URR555H n[ X5URUlM" gr )r)r;rrrr%rLwhichargnames rrlChLambda.on_changedsB5z..s499>>3C/DEG#*4#9DIIg  Frc L[R"U5Rn[U5VVs0sHupEU[ [ U5S9_M nnnUbUHnXv;dM X'XglM U"S0UD6n[UR55HsupYURU R 5(aURXYR 5 MBURRU5 [XU R 5 Mu X`lURSU5 gs snnf)Nr$_resultrT)r*rrrwrrr%rsrurDrr<r*r) rLrr argspec_argsrrr initial_argrargs rrMChLambda.__init__s--c277 AJ[US5(aSUl[URU]5$)Nr)r5rrlrr)rLrs rrChLambda.__getstate__s+ 4  DHT^^T799rc.URR$r rrbrs rrChLambda.compute_r||~~rc"XRLaggrG)rrs rrChLambda.compute_dr_wrts ,,  r)rrr )rOrPrQrRr<r>rxrlrMrrrrSrrs@rrrs: N #E F(J:*":rrc@\rSrSrSS/r/rSS/rSrSrSr Sr Sr g ) ChGroupirrc[U5R[URR555HGnURUR[ X5LdM*[ X5URUlMI gr )r)r;rrr%rrs rrlChGroup.on_changeds^5z..s499>>3C/DEG99W%''74+AA'.t'= '"$Frc [UR55VVs0sHup4U[US9_M snnUl[URR55Hup4[ XU5 UR UR 5(aURX4R 5 MNURRU5 [ XUR 5 M Xl gs snnf)Nr$) rsrurrrrDr%rr<r*r)rLrrrrs rrMChGroup.__init__ sCG CUWCU<7gx#.CUW  !23LG FS )&&w. !!'*suu- 4 XsC)c.URR$r rrs rrChGroup.compute_rrrc8URRU5$r )rrUrs rrChGroup.compute_dr_wrts||""3''r)rrN) rOrPrQrRr<r>rxrlrMrrrSrTrrrrs/ v E FF#J> (rr)*)__all__)r)linalg) ch_randomrrandomc(\rSrSr/SQrSrSrSrg)tsti0)rrccURRURR-URR-$r )rrbrrrs rr tst.compute_r3s*vvxx$&&(("TVVXX--rcgrGrTrs rrtst.compute_dr_wrt6srrTNrrTrrrr0s F.rrcV[n[S5n[S5n[S5n[S5n[UR UR 55 SSKoUR5 [XU5n[UR5 [U5 SSKoUR5 g)N cN[S5[S5[S5--S-$)Nr r)r)r%rzs rr&main..@s2a52a52a5=#81#rs_ E8II  % ++- //1$ --/x #6 #C;C;R"bkk&&() + +.< r &r&X(b(<):1>!Hh  " 8 zFr