U Zʹj<@sfddlZddlmZddlmZddlmZmZmZm 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@edZAZBeBrdddlCZCdeCjDd<e6dddZEe6dddZFe6dddZGe6dddZHed\ZIZJZKZLeAdk rddeMdddDZNeAOeNZPe.eNZQd;ddZRd&srD Fc Cs t||d}|s dd|D}ndd|D}t}d}|2dd|D}tjjj|d}|||}W5QRX|t t ||} | } | j rt | ts| } | } |s|| kstn:dd|D}dd| D} td d t || DstdS) Nr6cSsg|]}t|j|jqSrBr+rowscolsr@vrBrBrCrD.sz._compare_tensorflow_matrix..cSsg|]}t|j|jdqS)gY@rFrIrBrBrCrD0scSsg|]}tt|qSrBevalr1r?rBrBrCrD5sgraphcSsg|]}|D]}|q qSrBrBr@rowrArBrBrCrDCscSsg|]}|D]}|q qSrBrBrOrBrBrCrDDscss:|]2\}}t||ddttt|d kVqdS Nabsintr'r@abrBrBrC Esz-_compare_tensorflow_matrix..r3tfGraph as_defaultcompatv1SessionrunsubsdictzipdoitZ is_Matrix isinstancer)Z as_explicittolistallAssertionError variablesexpr use_floatfrandom_matricesrNrrandom_variablessessionerBrBrC_compare_tensorflow_matrix+s.   ruc Cs t||d}|s dd|D}ndd|D}t}d}|2dd|D}tjjj|d}|||}W5QRX|t t ||} | } | j rt | ts| } | } |s|| kstn:dd|D}dd| D} td d t || DstdS) Nr6cSsg|]}t|j|jdqS)rSr*rGrHrIrBrBrCrDNsz6_compare_tensorflow_matrix_inverse..cSsg|]}t|j|jdqS)gQ @rvrIrBrBrCrDPscSsg|]}tt|qSrBrKr?rBrBrCrDUsrMcSsg|]}|D]}|q qSrBrBrOrBrBrCrDcscSsg|]}|D]}|q qSrBrBrOrBrBrCrDdscss:|]2\}}t||ddttt|d kVqdSrQrTrWrBrBrCrZesz5_compare_tensorflow_matrix_inverse..r[rkrBrBrC"_compare_tensorflow_matrix_inverseKs.   rwc Cst||d}dd|D}t}d}|2dd|D}tjjj|d}|||}W5QRX|t t ||}| }t ||dkst dS)Nr6cSs"g|]}t|j|jdqS)d)r+rGrHevalfrIrBrBrCrDksz5_compare_tensorflow_matrix_scalar..cSsg|]}tt|qSrBrKr?rBrBrCrDqsrMư>)r3r\r]r^r_r`rarbrcrdrerfrUrj) rlrmrorprNrqrrrsrtrBrBrC!_compare_tensorflow_matrix_scalaris  r{cCs tddSNrrRrandomrandintrBrBrBrC{rc st||d}fdd|D}t}d}|2dd|D}tjjj|d}|||}W5QRX|t t || } t || dkstdS)Nr6csg|] }qSrBrBrIrngrBrCrD}sz._compare_tensorflow_scalar..cSsg|]}tt|qSrBrKr?rBrBrCrDsrMrz)r3r\r]r^r_r`rarbrcrdreryrfrUrj rlrmrroZrvsrNrqZtf_rvsrsrtrBrrC_compare_tensorflow_scalarzs  rcCs tddSr|r}rBrBrBrCrrc st||d}fdd|D}t}d}|2dd|D}tjjj|d}|||}W5QRX|t t || } || kst dS)Nr6csg|] }qSrBrBrIrrBrCrDsz2_compare_tensorflow_relational..cSsg|]}tt|qSrBrKr?rBrBrCrDsrM) r3r\r]r^r_r`rarbrcrdrerfrjrrBrrC_compare_tensorflow_relationals  rcCsFttddkstttttgttt gg}t|dksBtdS)Nr9z6tensorflow.constant([[1, 0, 0], [0, 1, 0], [0, 0, 1]])zPtensorflow.Variable([[x, tensorflow.math.sin(y)], [tensorflow.math.exp(z), -t]])) r1r*rjr(xryrztrmrBrBrCtest_tensorflow_printings rcCsts tdtt}t|dks$tttf|tt}t|dksHtttf|tt}t|dksltttf|dddt t}t|dkstttf|d ddt t}t|d kstttf|d ddt t}t|d kstttf|d ddtd}t|dkstttf|dddt t}t|dksLtttf|dddt t}t|dksztttf|dddtt}t|dkstttf|dddtt}t|dkstttf|dddtt}t|dkstttf|dddtt}t|dks2tttf|dddttt}t|dksbttttf|dddtt}t|dkstttf|d ddtt}t|d!kstttf|d"ddtt}t|d#kstttf|d$ddtt}t|d%kstttf|d&ddtt}t|d'ksJtttf|d(ddtt}t|d)ksxtttf|d*ddtt}t|d+kstttf|d,ddtt}t|d-kstttf|d.dddS)/NTensorFlow not installedztensorflow.math.abs(x)ztensorflow.math.sign(x)ztensorflow.math.ceil(x)cSstSNr~rBrBrBrCrrz&test_tensorflow_math..rztensorflow.math.floor(x)cSstSrrrBrBrBrCrrztensorflow.math.exp(x)cSstSrrrBrBrBrCrrztensorflow.math.sqrt(x)cSstSrrrBrBrBrCrrrSztensorflow.math.pow(x, 4)cSstSrrrBrBrBrCrrztensorflow.math.cos(x)cSstSrrrBrBrBrCrrztensorflow.math.acos(x)cSs tddS)Nrgffffff?r~uniformrBrBrBrCrrztensorflow.math.sin(x)cSstSrrrBrBrBrCrrztensorflow.math.asin(x)cSstSrrrBrBrBrCrrztensorflow.math.tan(x)cSstSrrrBrBrBrCrrztensorflow.math.atan(x)cSstSrrrBrBrBrCrrztensorflow.math.atan2(y, x)cSstSrrrBrBrBrCrrztensorflow.math.cosh(x)cSstSrrrBrBrBrCrrztensorflow.math.acosh(x)cSs tddSNrrBrBrBrCrrztensorflow.math.sinh(x)cSs tddSrrrBrBrBrCrrztensorflow.math.asinh(x)cSs tddSrrrBrBrBrCrrztensorflow.math.tanh(x)cSs tddSrrrBrBrBrCrrztensorflow.math.atanh(x)cSs tddS)Ngg?rrBrBrBrCrrztensorflow.math.erf(x)cSstSrrrBrBrBrCrrztensorflow.math.lgamma(x)cSstSrrrBrBrBrCrr)r\r4rrr1rjrrrrrrrrrrrrrrrrrrr r!r%r&rrBrBrCtest_tensorflow_maths   rcCs@tttdksttttdks(ttttdksacb", M, N)z ab,bc->acb)r\r4r]r^r_r`rar/rrr1rjr3rbZeinsumrirrZ transposer)rNrsr8r:r;r<rrZmcmdrrorrrBrBrCtest_codegen_extraxs            $ rcCsxtddd}tddd}tddd}t|ddks8ttd|ddksPt|d|||}t|d ksttdS) NArr9BC)rrzA[0, 0]z 3*A[0, 0]z&(tensorflow.math.add((-1)*B, A))[0, 0])r/r1rjrc)rrrFrBrBrCtest_MatrixElement_printings   rcCs"tttt}t|dkstdS)Nz2tensorflow.gradients(tensorflow.math.sin(x), x)[0])rrrr1rjrrBrBrCtest_tensorflow_Derivativesr)F)F)`r~Zsympy.core.functionrZsympy.core.symbolrZ0sympy.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.from_matrix_to_arrayr2Zsympy.utilities.lambdifyr3Zsympy.testing.pytestr4r5r\r6osenvironr8r:r;r<rrrrr>ZllorZm3x3Z m3x3sympyrurwr{rrrrrrrrrrrrBrBrBrCsX    l                  ` 5J