o 8Va<@sZddlZddlmZmZddlmZmZmZmZddl m Z m Z m Z m Z mZmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZm 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.m/Z/m0Z0ddl1m2Z2m3Z3m4Z4m5Z5m6Z6dd l7m8Z8dd l9m:Z:dd l;mZ>dd l=m?Z?edZ@ZAeArddlBZBdeBjCd<e5dddZDe5dddZEe5dddZFe5dddZGed\ZHZIZJZKe@durddeLdddDZMe@NeMZOe-eMZPd:ddZQd:ddZRdd ZSd!d"fd#d$ZTd%d"fd&d'ZUd(d)ZVe?d*d+ZWd,d-ZXd.d/ZYe?d0d1ZZd2d3Z[d4d5Z\d6d7Z]d8d9Z^dS);N)symbols Derivative)ArrayTensorProductArrayAdd PermuteDims ArrayDiagonal)EqNeGeGtLeLt) import_module)Absceilingexpfloorsignsinasinsqrtcosacostanatanatan2coshacoshsinhasinhtanhatanhreimargerfloggammalog)Matrix MatrixBaseeye randMatrix) DeterminantHadamardProductInverse MatrixSymbolTrace)tensorflow_code)convert_matrix_to_array)lambdify)skip)XFAIL tensorflow2ZTF_CPP_MIN_LOG_LEVELMNPQzx y z tcCs$g|]}ddt||dDqS)cSsg|]}|qSr=).0jr=r=F/usr/lib/python3/dist-packages/sympy/printing/tests/test_tensorflow.py %sz.r9)ranger>ir=r=r@rA%s$rA Fc C$t||d}|sdd|D}ndd|D}t}d}|dd|D}tjjj|d}|||}Wdn1sBwY|ddt ||D} | } | j rht | t sd| } | } |st|| ksrJdSd d|D}d d| D} td d t || DsJdS) Nr6cSsg|] }t|j|jqSr=r+rowscolsr>vr=r=r@rA-z._compare_tensorflow_matrix..cSsg|] }t|j|jdqS)gY@rGrJr=r=r@rA/cSg|]}tt|qSr=evalr1rCr=r=r@rA4graphcSi|]\}}||qSr=r=r>krKr=r=r@ 8z._compare_tensorflow_matrix..cSg|] }|D]}|qqSr=r=r>rowrDr=r=r@rABrLcSrYr=r=rZr=r=r@rACrLcs<|]\}}t||ddttt|d kVqdS Nabsintr'r>abr=r=r@ D, z-_compare_tensorflow_matrix..r3tfGraph as_defaultcompatv1SessionrunsubszipdoitZ is_Matrix isinstancer)Z as_explicittolistall variablesexpr use_floatfrandom_matricesrSrrandom_variablessessioner=r=r@_compare_tensorflow_matrix*0   rc CrF) Nr6cSg|] }t|j|jdqS)r_r*rHrIrJr=r=r@rAMrMz6_compare_tensorflow_matrix_inverse..cSr)gQ @rrJr=r=r@rAOrMcSrNr=rOrCr=r=r@rATrQrRcSrTr=r=rUr=r=r@rWXrXz6_compare_tensorflow_matrix_inverse..cSrYr=r=rZr=r=r@rAbrLcSrYr=r=rZr=r=r@rAcrLcsr\r]r`rcr=r=r@rfdrgz5_compare_tensorflow_matrix_inverse..rhrvr=r=r@"_compare_tensorflow_matrix_inverseJrrc Cst||d}dd|D}t}d}|dd|D}tjjj|d}|||}Wdn1s8wY|ddt ||D}| }t ||dksXJdS) Nr6cSs"g|] }t|j|jdqS)d)r+rHrIevalfrJr=r=r@rAjsz5_compare_tensorflow_matrix_scalar..cSrNr=rOrCr=r=r@rAprQrRcSrTr=r=rUr=r=r@rWtrXz5_compare_tensorflow_matrix_scalar..ư>) r3rirjrkrlrmrnrorprqrrra) rwrxrzr{rSr|r}r~rr=r=r@!_compare_tensorflow_matrix_scalarhs  rcC tddSNrr^randomZrandintr=r=r=r@z rc st||d}fdd|D}t}d}|dd|D}tjjj|d}|||}Wdn1s:wY|ddt ||D } t || dksZJdS) Nr6cg|]}qSr=r=rJrngr=r@rA|z._compare_tensorflow_scalar..cSrNr=rOrCr=r=r@rArQrRcSrTr=r=rUr=r=r@rWrXz._compare_tensorflow_scalar..r) r3rirjrkrlrmrnrorprqrrrra rwrxrrzZrvsrSr|Ztf_rvsr~rr=rr@_compare_tensorflow_scalarys  "rcCrrrr=r=r=r@rrc st||d}fdd|D}t}d}|dd|D}tjjj|d}|||}Wdn1s:wY|ddt ||D } || ksTJdS)Nr6crr=r=rJrr=r@rArz2_compare_tensorflow_relational..cSrNr=rOrCr=r=r@rArQrRcSrTr=r=rUr=r=r@rWrXz2_compare_tensorflow_relational..) r3rirjrkrlrmrnrorprqrrrr=rr@_compare_tensorflow_relationals  rcCsFttddks Jttttgttt gg}t|dks!JdS)Nr9z6tensorflow.constant([[1, 0, 0], [0, 1, 0], [0, 0, 1]])zPtensorflow.Variable([[x, tensorflow.math.sin(y)], [tensorflow.math.exp(z), -t]])) r1r*r(xryrztrxr=r=r@test_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)cStSNrr=r=r=r@rz&test_tensorflow_math..rztensorflow.math.floor(x)cSrrrr=r=r=r@rrztensorflow.math.exp(x)cSrrrr=r=r=r@rrztensorflow.math.sqrt(x)cSrrrr=r=r=r@rrr_ztensorflow.math.pow(x, 4)cSrrrr=r=r=r@rrztensorflow.math.cos(x)cSrrrr=r=r=r@rrztensorflow.math.acos(x)cSr)Nrgffffff?rZuniformr=r=r=r@rrztensorflow.math.sin(x)cSrrrr=r=r=r@rrztensorflow.math.asin(x)cSrrrr=r=r=r@rrztensorflow.math.tan(x)cSrrrr=r=r=r@rrztensorflow.math.atan(x)cSrrrr=r=r=r@rrztensorflow.math.atan2(y, x)cSrrrr=r=r=r@rrztensorflow.math.cosh(x)cSrrrr=r=r=r@rrztensorflow.math.acosh(x)cSrNrr=r=r=r@rrztensorflow.math.sinh(x)cSrrrr=r=r=r@rrztensorflow.math.asinh(x)cSrrrr=r=r=r@rrztensorflow.math.tanh(x)cSrrrr=r=r=r@rrztensorflow.math.atanh(x)cSr)Ngg?rr=r=r=r@rrztensorflow.math.erf(x)cSrrrr=r=r=r@rrztensorflow.math.lgamma(x)cSrrrr=r=r=r@rr)rir4rrr1rrrrrrrrrrrrrrrrrrr r!r%r&rr=r=r@test_tensorflow_maths       rcCs@tttdks JtttdksJtttdksJdS)Nztensorflow.math.real(x)ztensorflow.math.imag(x)ztensorflow.math.angle(x))r1r"rr#r$r=r=r=r@test_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)) rir4rrrr1rr r r r r rr=r=r@test_tensorflow_relational s(      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)Nrr8tensorflow.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)r9z;tensorflow.linalg.matmul(tensorflow.linalg.matmul(M, M), M)ztensorflow.linalg.trace(M)ztensorflow.linalg.det(M)ztensorflow.linalg.inv(M)T)ryz1.14)Ztensorflow_versionz%tensorflow.linalg.matrix_transpose(M)z1.13ztensorflow.matrix_transpose(M))rir4r8r1rr:r-r;r<r0r,rr.rTrr=r=r@test_tensorflow_matrices,sR       rc Cststdt}|Ztjjj|d}tddd}tddd}t||}t ||f|d}t ddgdd gg}t dd gd dgg}| |||}| t ||} || k s`JWddS1skwYdS) NrrRr8rr:r6rr9r_)rir4rjrkrlrmrnr/r2r3constantromatmulru) rSr~r8r:cgrzmambrcr=r=r@test_codegen_einsumbs    "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)Nrr8rr:r;r<rr9r_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]))rrr9rzKtensorflow.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)rir4rjrkrlrmrnr/rrr1r3roZeinsumrurrZ transposer)rSr~r8r:r;r<rrZmcZmdrrzrrr=r=r@test_codegen_extraws            $rcCsxtddd}tddd}tddd}t|ddksJtd|ddks(J|d|||}t|d ks:JdS) NArr9BC)rrzA[0, 0]z 3*A[0, 0]z&(tensorflow.math.add((-1)*B, A))[0, 0])r/r1rp)rrrFr=r=r@test_MatrixElement_printings   rcCs"tttt}t|dksJdS)Nz2tensorflow.gradients(tensorflow.math.sin(x), x)[0])rrrr1rr=r=r@test_tensorflow_Derivatives r)F)_rZsympyrrZ0sympy.tensor.array.expressions.array_expressionsrrrrZsympy.core.relationalrr r r r r Zsympy.externalrZsympy.functionsrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'Zsympy.matricesr(r)r*r+Zsympy.matrices.expressionsr,r-r.r/r0Zsympy.printing.tensorflowr1Z3sympy.tensor.array.expressions.conv_matrix_to_arrayr2Zsympy.utilities.lambdifyr3Zsympy.testing.pytestr4r5rir6osenvironr8r:r;r<rrrrrBZllorZm3x3Z m3x3sympyrrrrrrrrrrrrrrr=r=r=r@sX  l                 ` 5J