g<ddlZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z mZmZmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.m/Z/m0Z0m1Z1dd l2m3Z3m4Z4m5Z5m6Z6m7Z7dd l8m9Z9dd l:m;Z;dd lm?Z?ddl>m@Z@edxZAZBeBrddlCZCdeCjDd<e6dddZEe6dddZFe6dddZGe6dddZHed\ZIZJZKZLeA7deMdddDZNeAOeNZPe.eNZQd+dZRd+dZSdZTdfdZUd fd!ZVd"ZWe@d#ZXd$ZYd%ZZe@d&Z[d'Z\d(Z]d)Z^d*Z_dS),N) Derivative)symbols)ArrayTensorProductArrayAdd PermuteDims ArrayDiagonal)EqNeGeGtLeLt) import_module)Absceilingexpfloorsignsinasinsqrtcosacostanatanatan2coshacoshsinhasinhtanhatanhreimargerfloggammalog)Matrix MatrixBaseeye randMatrix) DeterminantHadamardProductInverse MatrixSymbolTrace)tensorflow_code)convert_matrix_to_array)lambdify)skip)XFAIL tensorflow2TF_CPP_MIN_LOG_LEVELMNPQzx y z tc Ng|]"}tt||dz#S)r;)listrange.0is Z/granary/edev/quantum/lib/python3.11/site-packages/sympy/printing/tests/test_tensorflow.py rF&s, 7 7 714a1   7 7 7 FcLt||d}|s d|D}n d|D}t}d}|5d|D}tjj|}|||}dddn #1swxYwY|tt||} | } | j r=t| ts| } | } |s|| ksJdSd|D}d| D} t#dt|| DsJdS) Nr7cBg|]}t|j|jSr,rowscolsrCvs rErFz._compare_tensorflow_matrix...s&III!:afaf55IIIrGcHg|]}t|j|jdz  S)gY@rLrOs rErFz._compare_tensorflow_matrix..0s+NNNq:afaf55d:NNNrGcFg|]}tt|SrKevalr2rBs rErFz._compare_tensorflow_matrix..5(NNND!3!344NNNrGgraphcg|] }|D]}| SrKrKrCrowrDs rErFz._compare_tensorflow_matrix..C% ) ) )3S ) )Q ) ) ) )rGcg|] }|D]}| SrKrKrYs rErFz._compare_tensorflow_matrix..Dr[rGc 3K|]L\}}t||z ddttt|dz zkVMdS Nabsintr(rCabs rE z-_compare_tensorflow_matrix..EqLL9=AC!HHrQs3s1vvr??33344 4LLLLLLrGr4tfGraph as_defaultcompatv1Sessionrunsubsdictzipdoit is_Matrix isinstancer* as_explicittolistall variablesexpr use_floatfrandom_matricesrWrrandom_variablessessiones rE_compare_tensorflow_matrixr+sD,//A OIIyIIINNINNN HHJJE A     ..NNoNNN),&&U&33 KK+, - -............... $s9o667788A A{!Z((  A HHJJ LQ||~~ ) )! ) ) ) ) )! ) ) )LLADQLLLLL L L L L LA B22B69B6cLt||d}|s d|D}n d|D}t}d}|5d|D}tjj|}|||}dddn #1swxYwY|tt||} | } | j r=t| ts| } | } |s|| ksJdSd|D}d| D} t#dt|| DsJdS) Nr7cHg|]}t|j|jdz S)r`r+rMrNrOs rErFz6_compare_tensorflow_matrix_inverse..Ns+DDDQ3qvqv..q0DDDrGcHg|]}t|j|jdz S)gQ @rrOs rErFz6_compare_tensorflow_matrix_inverse..Ps+GGG3qvqv..t3GGGrGcFg|]}tt|SrKrSrBs rErFz6_compare_tensorflow_matrix_inverse..UrUrGrVcg|] }|D]}| SrKrKrYs rErFz6_compare_tensorflow_matrix_inverse..cr[rGcg|] }|D]}| SrKrKrYs rErFz6_compare_tensorflow_matrix_inverse..dr[rGc 3K|]L\}}t||z ddttt|dz zkVMdSr^rards rErgz5_compare_tensorflow_matrix_inverse..erhrGrirzs rE"_compare_tensorflow_matrix_inverserKsD,//A HDD)DDDGGYGGG HHJJE A     ..NNoNNN),&&U&33 KK+, - -............... $s9o667788A A{!Z((  A HHJJ LQ||~~ ) )! ) ) ) ) )! ) ) )LLADQLLLLL L L L L Lrct||d}d|D}t}d}|5d|D}tjj|}|||}dddn #1swxYwY|tt||}| }t||z dksJdS)Nr7clg|]1}t|j|jdz 2S)d)r,rMrNevalfrOs rErFz5_compare_tensorflow_matrix_scalar..ksJEEE56 1616""((**S0EEErGcFg|]}tt|SrKrSrBs rErFz5_compare_tensorflow_matrix_scalar..qrUrGrVư>) r4rjrkrlrmrnrorprqrrrsrtrb) r{r|r~rrWrrrrs rE!_compare_tensorflow_matrix_scalarrisKD,//AEE:CEEEO HHJJE A     ..NNoNNN),&&U&33 KK+, - -............... $s9o667788A A qs88f      s A B##B'*B'c,tjddSNrr_randomrandintrKrGrEr{2!6!6rGc>t||d}fd|D}t}d}|5d|D}tjj|}|||}dddn #1swxYwY|tt|| } t|| z dksJdS)Nr7c$g|] } SrKrKrCrPrngs rErFz._compare_tensorflow_scalar..} $ $ $Q3355 $ $ $rGcFg|]}tt|SrKrSrBs rErFz._compare_tensorflow_scalar..(888q$q))**888rGrVr)r4rjrkrlrmrnrorprqrrrsrrtrb r{r|rr~rvsrWrtf_rvsrrs ` rE_compare_tensorflow_scalarrzsHD,//A $ $ $ $) $ $ $C HHJJE A     $$88C888),&&U&33 KK6 # #$$$$$$$$$$$$$$$ $s9c**++,,224499;;A qs88f      A B&&B*-B*c,tjddSrrrKrGrErrrrGct||d}fd|D}t}d}|5d|D}tjj|}|||}dddn #1swxYwY|tt|| } || ksJdS)Nr7c$g|] } SrKrKrs rErFz2_compare_tensorflow_relational..rrGcFg|]}tt|SrKrSrBs rErFz2_compare_tensorflow_relational..rrGrV) r4rjrkrlrmrnrorprqrrrsrtrs ` rE_compare_tensorflow_relationalrs,D,//A $ $ $ $) $ $ $C HHJJE A     $$88C888),&&U&33 KK6 # #$$$$$$$$$$$$$$$ $s9c**++,,1133A 666666rcttddksJttt t gt tt gg}t|dksJdS)Nr;z6tensorflow.constant([[1, 0, 0], [0, 1, 0], [0, 0, 1]])zPtensorflow.Variable([[x, tensorflow.math.sin(y)], [tensorflow.math.exp(z), -t]])) r2r+r)xryrztr|s rEtest_tensorflow_printingrs 3q66 " "@ A A A A As1vv;Q! - . .D 4  . . . . . . .rGc tstdtt}t |dksJt tf|t t}t |dksJt tf|tt}t |dksJt tf|dtt}t |dksJt tf|dtt}t |d ksJt tf|d tt}t |d ksJt tf|d td z}t |dksJt tf|dtt}t |dksJt tf|dtt}t |dksJt tf|dtt}t |dksJt tf|dtt}t |dksJt tf|dtt}t |dksJt tf|dt!t}t |dksJt tf|dt#t$t}t |dksJt t$tf|dt't}t |dksJt tf|dt)t}t |d ksJt tf|d!t+t}t |d"ksJt tf|d#t-t}t |d$ksJt tf|d%t/t}t |d&ksJt tf|d't1t}t |d(ksJt tf|d)t3t}t |d*ksJt tf|d+t5t}t |d,ksJt tf|d-dS).NTensorFlow not installedztensorflow.math.abs(x)ztensorflow.math.sign(x)ztensorflow.math.ceil(x)c(tjSNrrKrGrErz&test_tensorflow_math.. v}rG)rztensorflow.math.floor(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.exp(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.sqrt(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGr`ztensorflow.math.pow(x, 4)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.cos(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.acos(x)c,tjddS)Nrgffffff?runiformrKrGrErz&test_tensorflow_math..sv~a7N7NrGztensorflow.math.sin(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.asin(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.tan(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.atan(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.atan2(y, x)c(tjSrrrKrGrErz&test_tensorflow_math..s rGztensorflow.math.cosh(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.acosh(x)c,tjddSNrrKrGrErz&test_tensorflow_math..v~a7K7KrGztensorflow.math.sinh(x)c,tjddSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.asinh(x)c,tjddSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.tanh(x)c,tjddSrrrKrGrErz&test_tensorflow_math..rrGztensorflow.math.atanh(x)c,tjddS)Ngg?rrKrGrErz&test_tensorflow_math..ssB 7 7rGztensorflow.math.erf(x)c(tjSrrrKrGrErz&test_tensorflow_math..  rGztensorflow.math.lgamma(x)c(tjSrrrKrGrErz&test_tensorflow_math..rrG)rjr5rrr2rrrrrrrrrrrrrrrrrr r!r"r&r'rs rEtest_tensorflow_mathrs ) '((( q66D 4 $< < < < <tT*** 77D 4 $= = = = =tT*** 1::D 4 $= = = = =tT/F/FGGGG 88D 4 $> > > > >tT/F/FGGGG q66D 4 $< < < < <tT/F/FGGGG 77D 4 $= = = = =tT/F/FGGGG 6D 4 $? ? ? ? ?tT/F/FGGGG q66D 4 $< < < < <tT/F/FGGGG 77D 4 $= = = = =tT/N/NOOOO q66D 4 $< < < < <tT/F/FGGGG 77D 4 $= = = = =tT/F/FGGGG q66D 4 $< < < < <tT/F/FGGGG 77D 4 $= = = = =tT/F/FGGGG A;;D 4 $A A A A A1vt1H1HIIII 77D 4 $= = = = =tT/F/FGGGG 88D 4 $> > > > >tT/K/KLLLL 77D 4 $= = = = =tT/K/KLLLL 88D 4 $> > > > >tT/K/KLLLL 77D 4 $= = = = =tT/K/KLLLL 88D 4 $> > > > > d779999 q66D 4 $< < < < < d//1111 A;;D 4 $? ? ? ? ? d//111111rGctttdksJtttdksJtt tdksJdS)Nztensorflow.math.real(x)ztensorflow.math.imag(x)ztensorflow.math.angle(x))r2r#rr$r%rKrGrEtest_tensorflow_complexesrsh 2a55 ! !%> > > > > 2a55 ! !%> > > > > 3q66 " "&@ @ @ @ @ @ @rGctstdttt}t |dksJt ttf|ttt}t |dksJt ttf|ttt}t |dksJt ttf|ttt}t |dksJt ttf|ttt}t |dksJt ttf|ttt}t |dksJt ttf|dS)Nrztensorflow.math.equal(x, y)ztensorflow.math.not_equal(x, y)z#tensorflow.math.greater_equal(x, y)ztensorflow.math.greater(x, y)z tensorflow.math.less_equal(x, y)ztensorflow.math.less(x, y)) rjr5r rrr2rr r r r rrs rEtest_tensorflow_relationalrs ) '((( a88D 4 $A A A A A"Aq64000 a88D 4 $E E E E E"Aq64000 a88D 4 $I I I I I"Aq64000 a88D 4 $C C C C C"Aq64000 a88D 4 $F F F F F"Aq64000 a88D 4 $@ @ @ @ @"Aq6400000rGcltstdt}t|dksJt tf|tt z}t|dksJt tt f|tt z}t|dksJt tt f|t tt }t|dksJt tt f|tt ztztz}t|dksJt tt ttf|tdz}t|dksJt tf|tt}t|d ksJt tf|tt}t|d ksJttf|tt}t|d ksJttf|d tj}t|ddksJt|ddksJt tf|dS)Nrr:tensorflow.math.add(M, N)ztensorflow.linalg.matmul(M, N)ztensorflow.math.multiply(M, N)zXtensorflow.linalg.matmul(tensorflow.linalg.matmul(tensorflow.linalg.matmul(M, N), P), Q)r;z;tensorflow.linalg.matmul(tensorflow.linalg.matmul(M, M), M)ztensorflow.linalg.trace(M)ztensorflow.linalg.det(M)ztensorflow.linalg.inv(M)T)r}z1.14)tensorflow_versionz%tensorflow.linalg.matrix_transpose(M)z1.13ztensorflow.matrix_transpose(M))rjr5r:r2rr<r.r=r>r1r-rr/rTrs rEtest_tensorflow_matricesr-s ) '((( D 4 C ' ' ' 'tT*** q5D 4 $? ? ? ? ?1vt,,, q5D 4 $D D D D D1vt,,, 1a D 4 $D D D D D1vt,,, Q3q57D 4  9 9 9 9 91a|T222 a4D 4 E F F F FtT*** 88D 4 $@ @ @ @ @tT*** q>>D 4 $> > > > >%qdD111 1::D 4 $> > > > >&tTTBBBB 3D 4F ; ; ;/ 0 0 0 0 4F ; ; ;( ) ) ) )tT*****rGctstdt}|5tjj|}tddd}tddd}t||z}t||f|d}t ddgdd gg}t dd gd dgg}| |||}| t ||} || k sJ ddddS#1swxYwYdS) NrrVr:rr<r7rr;r`)rjr5rkrlrmrnror0r3r4constantrpmatmulry) rWrr:r<cgr~mambrcs rEtest_codegen_einsumrcs ) '((( HHJJE       ),&&U&33 a # # a # # $QU + + aVR . . [[1a&1a&) * * [[1R&2q'* + + KK"b " " KK "b)) * *Q||~~                  sDE&&E*-E*c V tstdt}|5tjj}tddd}tddd}tddd}tddd}tddgdd gg}tdd gd dgg}tdd gddgg}tdd gd d gg} t||} t| dksJt||f| d} | | ||} | t d||} | | ksJt||} t| dksJt||f| d} | | ||} | ||z} | | ksJt|||} t| dksJt|||f| d} | | |||} | ||z|z} | | ksJt||||} t| dksJt||||f| d} | | |||| } | ||z|z| z} | | ksJt!|dd g} t| dksJt|f| d} | | |} | t|} | | ksJt!t||gd} t| dksJt||f| d} | | ||} | tt d||gd} | | ksJt%t||d} t| dksJt||f| d} | | ||} | t d||} | | ksJ ddddS#1swxYwYdS)Nrr:rr<r=r>rr;r`rrrz'tensorflow.linalg.einsum("ab,cd", M, N)r7zij,klrz1tensorflow.math.add(tensorflow.math.add(M, N), P)zItensorflow.math.add(tensorflow.math.add(tensorflow.math.add(M, N), P), Q)ztensorflow.transpose(M, [1, 0]))rrr;rzKtensorflow.transpose(tensorflow.linalg.einsum("ab,cd", M, N), [1, 2, 3, 0])zab,cd)rrz,tensorflow.linalg.einsum("ab,bc->acb", M, N)z ab,bc->acb)rjr5rkrlrmrnror0rrr2r4rpeinsumryrr transposer)rWrr:r<r=r>rrmcmdrr~rrs rEtest_codegen_extrarxs ) '((( HHJJE     BB),&&(( a # # a # # a # # a # # [[1a&1a&) * * [[1R&2q'* + + [[1a&1a&) * * [[1R&1a&) * * 1 % %r"" 56666 aVR . . KK"b " " KK '2r22 3 3Q||~~ a^^r""&AAAAA aVR . . KK"b " " KKR Q||~~ aA  r"" ?@@@@ aAYL 1 1 KK"b" & & KKR" % %Q||~~ aAq ! !r"" HHHHH aAq\2| 4 4 KK"b"b)) * * KKR" r) * *Q||~~ QF # #r""&GGGGG aT2| , , KK"   KK R(( ) )Q||~~ +Aq11<<< @ @r"" IIIII aVR . . KK"b " " KK RYYwB%?%?NN O OQ||~~ -a33V < <r"" :;;;; aVR . . KK"b " " KK ,B77 8 8Q||~~EBBBBBBBBBBBBBBBBBBsU VV"%V"cFtddd}tddd}tddd}t|ddksJtd|dzdksJ|d|||z }t|d ksJdS) NArr;BC)rrzA[0, 0]z 3*A[0, 0]z&(tensorflow.math.add((-1)*B, A))[0, 0])r0r2rq)rrrFs rEtest_MatrixElement_printingrsS!QAS!QAS!QA 1T7 # #y 0 0 0 0 1qw; ' '; 6 6 6 6 $ QAA 1  !I I I I I I IrGc~tttt}t|dksJdS)Nz2tensorflow.gradients(tensorflow.math.sin(x), x)[0])rrrr2rs rEtest_tensorflow_DerivativersC c!ffa D 4 < = = = = = =rG)F)`rsympy.core.functionrsympy.core.symbolr0sympy.tensor.array.expressions.array_expressionsrrrrsympy.core.relationalr r r r r rsympy.externalrsympy.functionsrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(sympy.matricesr)r*r+r,sympy.matrices.expressionsr-r.r/r0r1sympy.printing.tensorflowr23sympy.tensor.array.expressions.from_matrix_to_arrayr3sympy.utilities.lambdifyr4sympy.testing.pytestr5r6rjr7osenvironr:r<r=r>rrrrrAllorm3x3 m3x3sympyrrrrrrrrrrrrrrrKrGrEr s ******%%%%%%8888888888888888(((((($$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$?>>>>>>>>>>>??????????????555555WWWWWW------%%%%%%&&&&&& - ---Z - III),BJ%&LaLaLaLa WY   1a> 7 7aA 7 7 7C ;;s  Ds ILLLL@LLLL<$76    "76     ...]1]1]1@AAA 111>2+2+2+j*GGGT J J J=====rG