o g<@sbddlZddlmZddlmZddlmZmZmZm 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@edZAZBeBrddlCZCdeCjDd<e6dddZEe6dddZFe6dddZGe6dddZHed\ZIZJZKZLeAdurddeMdddDZNeAOeNZPe.eNZQd;ddZRd;ddZSd d!ZTd"d#fd$d%ZUd&d#fd'd(ZVd)d*ZWe@d+d,ZXd-d.ZYd/d0ZZe@d1d2Z[d3d4Z\d5d6Z]d7d8Z^d9d: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 tcCsg|] }tt||dqS)r:)listrange.0irC[/var/www/visachat/venv/lib/python3.10/site-packages/sympy/printing/tests/test_tensorflow.py &rE Fc Ct||d}|sdd|D}ndd|D}t}d}|dd|D}tjjj|d}|||}Wdn1sBwY|t t ||} | } | j ret | tsa| } | } |sq|| ksoJdSdd|D}dd| D} td d t || DsJdS) Nr6cSsg|] }t|j|jqSrCr+rowscolsrAvrCrCrDrE.z._compare_tensorflow_matrix..cSsg|] }t|j|jdqS)gY@rIrLrCrCrDrE0rFcSg|]}tt|qSrCevalr1r@rCrCrDrE5graphcSg|] }|D]}|qqSrCrCrArowrBrCrCrDrECrNcSrUrCrCrVrCrCrDrEDrNcs<|]\}}t||ddttt|d kVqdS Nabsintr'rAabrCrCrD E, z-_compare_tensorflow_matrix..r3tfGraph as_defaultcompatv1Sessionrunsubsdictzipdoit is_Matrix isinstancer) as_explicittolistall variablesexpr use_floatfrandom_matricesrTrrandom_variablessessionerCrCrD_compare_tensorflow_matrix+0   rc CrH) Nr6cSg|] }t|j|jdqS)r[r*rJrKrLrCrCrDrENrFz6_compare_tensorflow_matrix_inverse..cSr)gQ @rrLrCrCrDrEPrFcSrOrCrPr@rCrCrDrEUrRrScSrUrCrCrVrCrCrDrEcrNcSrUrCrCrVrCrCrDrEdrNcsrXrYr\r_rCrCrDrbercz5_compare_tensorflow_matrix_inverse..rdrurCrCrD"_compare_tensorflow_matrix_inverseKrrc Cst||d}dd|D}t}d}|dd|D}tjjj|d}|||}Wdn1s8wY|t t ||}| }t ||dksUJdS)Nr6cSs"g|] }t|j|jdqS)d)r+rJrKevalfrLrCrCrDrEksz5_compare_tensorflow_matrix_scalar..cSrOrCrPr@rCrCrDrEqrRrSư>) r3rerfrgrhrirjrkrlrmrnror]) rvrwryrzrTr{r|r}r~rCrCrD!_compare_tensorflow_matrix_scalaris  rcC tddSNrrZrandomrandintrCrCrCrD{ rc st||d}fdd|D}t}d}|dd|D}tjjj|d}|||}Wdn1s:wY|t t || } t || dksWJdS)Nr6cg|]}qSrCrCrLrngrCrDrE}z._compare_tensorflow_scalar..cSrOrCrPr@rCrCrDrErRrSr)r3rerfrgrhrirjrkrlrmrnrror] rvrwrryrvsrTr{tf_rvsr}r~rCrrD_compare_tensorflow_scalarzs  rcCrrrrCrCrCrDrrc st||d}fdd|D}t}d}|dd|D}tjjj|d}|||}Wdn1s:wY|t t || } || ksQJdS)Nr6crrCrCrLrrCrDrErz2_compare_tensorflow_relational..cSrOrCrPr@rCrCrDrErRrS) r3rerfrgrhrirjrkrlrmrnrorrCrrD_compare_tensorflow_relationals  rcCsFttddks Jttttgttt gg}t|dks!JdS)Nr:z6tensorflow.constant([[1, 0, 0], [0, 1, 0], [0, 0, 1]])zPtensorflow.Variable([[x, tensorflow.math.sin(y)], [tensorflow.math.exp(z), -t]])) r1r*r(xryrztrwrCrCrDtest_tensorflow_printings  rcCststdtt}t|dksJttf|tt}t|dks$Jttf|tt}t|dks6Jttf|dddtt}t|dksLJttf|d ddt t}t|d ksbJttf|d ddt t}t|d ksxJttf|d ddtd}t|dksJttf|dddt t}t|dksJttf|dddt t}t|dksJttf|dddt t}t|dksJttf|dddtt}t|dksJttf|dddtt}t|dksJttf|dddtt}t|dksJttf|dddttt}t|dks+Jtttf|dddtt}t|dksCJttf|d ddtt}t|d!ksZJttf|d"ddtt}t|d#ksqJttf|d$ddtt}t|d%ksJttf|d&ddtt}t|d'ksJttf|d(ddtt}t|d)ksJttf|d*ddtt}t|d+ksJttf|d,ddtt}t|d-ksJttf|d.dddS)/NTensorFlow not installedztensorflow.math.abs(x)ztensorflow.math.sign(x)ztensorflow.math.ceil(x)cStSNrrCrCrCrDrz&test_tensorflow_math..rztensorflow.math.floor(x)cSrrrrCrCrCrDrrztensorflow.math.exp(x)cSrrrrCrCrCrDrrztensorflow.math.sqrt(x)cSrrrrCrCrCrDrrr[ztensorflow.math.pow(x, 4)cSrrrrCrCrCrDrrztensorflow.math.cos(x)cSrrrrCrCrCrDrrztensorflow.math.acos(x)cSr)Nrgffffff?runiformrCrCrCrDrrztensorflow.math.sin(x)cSrrrrCrCrCrDrrztensorflow.math.asin(x)cSrrrrCrCrCrDrrztensorflow.math.tan(x)cSrrrrCrCrCrDrrztensorflow.math.atan(x)cSrrrrCrCrCrDrrztensorflow.math.atan2(y, x)cSrrrrCrCrCrDrrztensorflow.math.cosh(x)cSrrrrCrCrCrDrrztensorflow.math.acosh(x)cSrNrrCrCrCrDrrztensorflow.math.sinh(x)cSrrrrCrCrCrDrrztensorflow.math.asinh(x)cSrrrrCrCrCrDrrztensorflow.math.tanh(x)cSrrrrCrCrCrDrrztensorflow.math.atanh(x)cSr)Ngg?rrCrCrCrDrrztensorflow.math.erf(x)cSrrrrCrCrCrDrrztensorflow.math.lgamma(x)cSrrrrCrCrCrDrr)rer4rrr1rrrrrrrrrrrrrrrrrrr r!r%r&rrCrCrDtest_tensorflow_maths       rcCs@tttdks JtttdksJtttdksJdS)Nztensorflow.math.real(x)ztensorflow.math.imag(x)ztensorflow.math.angle(x))r1r"rr#r$rCrCrCrDtest_tensorflow_complexessrcCststdttt}t|dksJtttf|ttt}t|dks'Jtttf|ttt}t|dks;Jtttf|t tt}t|dksOJtttf|t tt}t|dkscJtttf|t tt}t|dkswJtttf|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)) rer4rrrr1rr r r r r rrCrCrDtest_tensorflow_relationals(      rcCststdt}t|dksJttf|tt}t|dks"Jtttf|tt}t|dks5Jtttf|ttt}t|dksIJtttf|tttt}t|dks`Jtttttf|td}t|dksuJttf|t t}t|d ksJttf|t t}t|d ksJt tf|t t}t|d ksJt tf|d d tj}t|dddksJt|dddksJttf|dS)Nrr9tensorflow.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)rxz1.14)tensorflow_versionz%tensorflow.linalg.matrix_transpose(M)z1.13ztensorflow.matrix_transpose(M))rer4r9r1rr;r-r<r=r0r,rr.rTrrCrCrDtest_tensorflow_matrices-sR       rc Cststdt}|Ztjjj|d}tddd}tddd}t||}t ||f|d}t ddgdd gg}t dd gd dgg}| |||}| t ||} || k s`JWddS1skwYdS) NrrSr9rr;r6rr:r[)rer4rfrgrhrirjr/r2r3constantrkmatmulrt) rTr}r9r;cgrymambrcrCrCrDtest_codegen_einsumcs    "rc Cststdt}|tjj}tddd}tddd}tddd}tddd}tddgdd gg}tdd gd dgg}tdd gddgg}tdd gd d gg} t ||} t | dksgJt ||f| d} | | ||} | t d||} | | ksJt||} t | dksJt ||f| d} | | ||} | ||} | | ksJt|||} t | dksJt |||f| d} | | |||} | |||} | | ksJt||||} t | dksJt ||||f| d} | | |||| } | |||| } | | ksJt|dd g} t | dks-Jt |f| d} | | |} | t|} | | ksLJtt ||gd} t | dks_Jt ||f| d} | | ||} | tt d||gd} | | ksJtt ||d} t | dksJt ||f| d} | | ||} | t d||} | | ksJWddS1swYdS)Nrr9rr;r<r=rr:r[rrrz'tensorflow.linalg.einsum("ab,cd", M, N)r6zij,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)rer4rfrgrhrirjr/rrr1r3rkeinsumrtrr transposer)rTr}r9r;r<r=rrmcmdrryrrrCrCrDtest_codegen_extraxs            $rcCsxtddd}tddd}tddd}t|ddksJtd|ddks(J|d|||}t|d ks:JdS) NArr:BC)rrzA[0, 0]z 3*A[0, 0]z&(tensorflow.math.add((-1)*B, A))[0, 0])r/r1rl)rrrFrCrCrDtest_MatrixElement_printings   rcCs"tttt}t|dksJdS)Nz2tensorflow.gradients(tensorflow.math.sin(x), x)[0])rrrr1rrCrCrDtest_tensorflow_Derivatives r)F)`rsympy.core.functionrsympy.core.symbolr0sympy.tensor.array.expressions.array_expressionsrrrrsympy.core.relationalrr r r r r sympy.externalrsympy.functionsrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'sympy.matricesr(r)r*r+sympy.matrices.expressionsr,r-r.r/r0sympy.printing.tensorflowr13sympy.tensor.array.expressions.from_matrix_to_arrayr2sympy.utilities.lambdifyr3sympy.testing.pytestr4r5rer6osenvironr9r;r<r=rrrrr?llorm3x3 m3x3sympyrrrrrrrrrrrrrrrCrCrCrDsZ    l                 ` 5J