o
    hB                     @   s  d dl Z d dlmZ d dlmZ d dlmZ d dlmZm	Z	m
Z
mZ d dlmZmZmZmZmZmZ d dlmZ d dlm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"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. d d	l/m0Z0m1Z1 d d
l2m3Z3m4Z4m5Z5m6Z6 d dl7m8Z8m9Z9m:Z:m;Z;m<Z< d dl=m>Z> d dl?m@Z@ d dlAmBZB d dlCmDZD d dlCmEZE ed ZFZGeGrd dlHZHdeHjId< e;dddZJe;dddZKe;dddZLe;dddZMed\ZNZOZPZQeFdurdd eRd ddD ZSeFTeSZUe3eSZVd?ddZWd?d d!ZXd"d# ZYd$d% fd&d'ZZd(d% fd)d*Z[d+d, Z\eEd-d. Z]d/d0 Z^d1d2 Z_eEd3d4 Z`d5d6 Zad7d8 Zbd9d: Zcd;d< Zdd=d> ZedS )@    N)
Derivative)symbols)	Piecewise)ArrayTensorProductArrayAddPermuteDimsArrayDiagonal)EqNeGeGtLeLt)import_module)Absceilingexpfloorsignsinasinsqrtcosacostanatanatan2coshacoshsinhasinhtanhatanhreimargerfloggammalog)isnanisinf)Matrix
MatrixBaseeye
randMatrix)DeterminantHadamardProductInverseMatrixSymbolTrace)tensorflow_code)convert_matrix_to_array)lambdify)skip)XFAIL
tensorflow2TF_CPP_MIN_LOG_LEVELM   NPQzx y z tc                 C   s   g | ]}t t||d  qS )r=   )listrange.0i rF   Z/var/www/html/ai/venv/lib/python3.10/site-packages/sympy/printing/tests/test_tensorflow.py
<listcomp>(       rH   	   Fc           
      C     t | |d}|sdd | D }ndd | D }t }d }|  dd |D }tjjj|d}||| }W d    n1 sBw   Y  |t	t
| |}	|	 }	|	jret|	tsa|	 }	|	 }	|sq||	k soJ d S dd |D }dd |	D }	td	d
 t
||	D sJ d S )Nr9   c                 S   s   g | ]	}t |j|jqS rF   r.   rowscolsrD   vrF   rF   rG   rH   0       z._compare_tensorflow_matrix.<locals>.<listcomp>c                 S   s   g | ]}t |j|jd  qS )g      Y@rL   rO   rF   rF   rG   rH   2   rI   c                 S      g | ]}t t|qS rF   evalr4   rC   rF   rF   rG   rH   7       graphc                 S      g | ]	}|D ]}|qqS rF   rF   rD   rowrE   rF   rF   rG   rH   E   rQ   c                 S   rX   rF   rF   rY   rF   rF   rG   rH   F   rQ   c                 s   <    | ]\}}t || d dttt |d    k V  qdS 
      Nabsintr(   rD   abrF   rF   rG   	<genexpr>G       ,
z-_compare_tensorflow_matrix.<locals>.<genexpr>r6   tfGraph
as_defaultcompatv1Sessionrunsubsdictzipdoit	is_Matrix
isinstancer,   as_explicittolistall
	variablesexpr	use_floatfrandom_matricesrW   rrandom_variablessessionerF   rF   rG   _compare_tensorflow_matrix-   0   

r   c           
      C   rK   )Nr9   c                 S      g | ]}t |j|jd  qS )r^   r-   rM   rN   rO   rF   rF   rG   rH   P   rI   z6_compare_tensorflow_matrix_inverse.<locals>.<listcomp>c                 S   r   )gQ	@r   rO   rF   rF   rG   rH   R   rI   c                 S   rR   rF   rS   rC   rF   rF   rG   rH   W   rU   rV   c                 S   rX   rF   rF   rY   rF   rF   rG   rH   e   rQ   c                 S   rX   rF   rF   rY   rF   rF   rG   rH   f   rQ   c                 s   r[   r\   r_   rb   rF   rF   rG   re   g   rf   z5_compare_tensorflow_matrix_inverse.<locals>.<genexpr>rg   rx   rF   rF   rG   "_compare_tensorflow_matrix_inverseM   r   r   c           	      C   s   t | |d}dd | D }t }d }|  dd |D }tjjj|d}||| }W d    n1 s8w   Y  |t	t
| |}| }t|| dk sUJ d S )Nr9   c                 S   s"   g | ]}t |j|j d  qS )d   )r.   rM   rN   evalfrO   rF   rF   rG   rH   m   s    z5_compare_tensorflow_matrix_scalar.<locals>.<listcomp>c                 S   rR   rF   rS   rC   rF   rF   rG   rH   s   rU   rV   ư>)r6   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   r`   )	ry   rz   r|   r}   rW   r~   r   r   r   rF   rF   rG   !_compare_tensorflow_matrix_scalark   s   
r   c                   C      t ddS Nr   r]   randomrandintrF   rF   rF   rG   <lambda>}       r   c           
         s   t | |d} fdd| D }t }d }|  dd |D }tjjj|d}||| }W d    n1 s:w   Y  |t	t
| |  }	t||	 dk sWJ d S )Nr9   c                       g | ]}  qS rF   rF   rO   rngrF   rG   rH          z._compare_tensorflow_scalar.<locals>.<listcomp>c                 S   rR   rF   rS   rC   rF   rF   rG   rH      rU   rV   r   )r6   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   r   rr   r`   
ry   rz   r   r|   rvsrW   r~   tf_rvsr   r   rF   r   rG   _compare_tensorflow_scalar|   s   
r   c                   C   r   r   r   rF   rF   rF   rG   r      r   c           
         s   t | |d} fdd| D }t }d }|  dd |D }tjjj|d}||| }W d    n1 s:w   Y  |t	t
| | }	||	ksQJ d S )Nr9   c                    r   rF   rF   rO   r   rF   rG   rH      r   z2_compare_tensorflow_relational.<locals>.<listcomp>c                 S   rR   rF   rS   rC   rF   rF   rG   rH      rU   rV   )r6   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   r   rF   r   rG   _compare_tensorflow_relational   s   
r   c                  C   sF   t tddks
J ttttgttt gg} t | dks!J d S )Nr=   z6tensorflow.constant([[1, 0, 0], [0, 1, 0], [0, 0, 1]])zPtensorflow.Variable([[x, tensorflow.math.sin(y)], [tensorflow.math.exp(z), -t]]))	r4   r-   r+   xr   yr   ztrz   rF   rF   rG   test_tensorflow_printing   s   
r   c                  C   s  t std tt} t| dksJ ttf|  tt} t| dks$J ttf|  tt} t| dks6J ttf| dd d tt} t| dksLJ ttf| d	d d t	t} t| d
ksbJ ttf| dd d t
t} t| dksxJ ttf| dd d td } t| dksJ ttf| dd d tt} t| dksJ ttf| dd d tt} t| dksJ ttf| dd d tt} t| dksJ ttf| dd d tt} t| dksJ ttf| dd d tt} t| dksJ ttf| dd d tt} t| dksJ ttf| dd d ttt} t| dks+J tttf| dd d tt} t| dksCJ ttf| d d d tt} t| d!ksZJ ttf| d"d d tt} t| d#ksqJ ttf| d$d d tt} t| d%ksJ ttf| d&d d tt} t| d'ksJ ttf| d(d d tt} t| d)ksJ ttf| d*d d tt} t| d+ksJ ttf| d,d d tt} t| d-ksJ ttf| d.d d d S )/NTensorFlow not installedztensorflow.math.abs(x)ztensorflow.math.sign(x)ztensorflow.math.ceil(x)c                   S      t   S Nr   rF   rF   rF   rG   r          z&test_tensorflow_math.<locals>.<lambda>r   ztensorflow.math.floor(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.exp(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.sqrt(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   r^   ztensorflow.math.pow(x, 4)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.cos(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.acos(x)c                   S   r   )Nr   gffffff?r   uniformrF   rF   rF   rG   r      r   ztensorflow.math.sin(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.asin(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.tan(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.atan(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.atan2(y, x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.cosh(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.acosh(x)c                   S   r   N      r   rF   rF   rF   rG   r      r   ztensorflow.math.sinh(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.asinh(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.tanh(x)c                   S   r   r   r   rF   rF   rF   rG   r      r   ztensorflow.math.atanh(x)c                   S   r   )Ng      g      ?r   rF   rF   rF   rG   r      r   ztensorflow.math.erf(x)c                   S   r   r   r   rF   rF   rF   rG   r     r   ztensorflow.math.lgamma(x)c                   S   r   r   r   rF   rF   rF   rG   r     r   )rh   r7   r   r   r4   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r&   r'   r   rF   rF   rG   test_tensorflow_math   s   

r   c                   C   s@   t ttdks
J t ttdksJ t ttdksJ d S )Nztensorflow.math.real(x)ztensorflow.math.imag(x)ztensorflow.math.angle(x))r4   r#   r   r$   r%   rF   rF   rF   rG   test_tensorflow_complexes
  s   r   c                  C   s   t std ttt} t| dksJ tttf|  ttt} t| dks'J tttf|  ttt} t| dks;J tttf|  t	tt} t| dksOJ tttf|  t
tt} t| dkscJ tttf|  ttt} t| dkswJ tttf|  d S )Nr   ztensorflow.math.equal(x, y)ztensorflow.math.not_equal(x, y)z#tensorflow.math.greater_equal(x, y)ztensorflow.math.greater(x, y)z tensorflow.math.less_equal(x, y)ztensorflow.math.less(x, y))rh   r7   r	   r   r   r4   r   r
   r   r   r   r   r   rF   rF   rG   test_tensorflow_relational  s(   





r   c                  C   s  t std t} t| dksJ ttf|  tt } t| dks"J tttf|  tt } t| dks5J tttf|  ttt} t| dksIJ tttf|  tt t t } t| dks`J tttttf|  td } t| dksuJ ttf|  t	t} t| d	ksJ ttf|  t
t} t| d
ksJ ttf|  tt} t| dksJ ttf| dd tj} t| dddksJ t| dddksJ ttf|  d S )Nr   r<   tensorflow.math.add(M, N)ztensorflow.linalg.matmul(M, N)ztensorflow.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)ztensorflow.linalg.trace(M)ztensorflow.linalg.det(M)ztensorflow.linalg.inv(M)T)r{   z1.14)tensorflow_versionz%tensorflow.linalg.matrix_transpose(M)z1.13ztensorflow.matrix_transpose(M))rh   r7   r<   r4   r   r>   r0   r?   r@   r3   r/   r   r1   r   Tr   rF   rF   rG   test_tensorflow_matrices/  sR   


r   c            
      C   s   t std t  } |  Z t jjj| d}tddd}tddd}t|| }t	||f|d}t 
ddgdd	gg}t 
dd
gddgg}||||}|t ||}	||	k s`J W d    d S 1 skw   Y  d S )Nr   rV   r<   r   r>   r9   r   r=   r^   )rh   r7   ri   rj   rk   rl   rm   r2   r5   r6   constantrn   matmulrw   )
rW   r   r<   r>   cgr|   mambr   crF   rF   rG   test_codegen_einsume  s   
"r   c               
   C   s  t std t  } |   t jj }tddd}tddd}tddd}tddd}t ddgdd	gg}t dd
gddgg}t ddgddgg}t ddgd	dgg}	t	||}
t
|
dksgJ t||f|
d}||||}|t d||}||k sJ t||}
t
|
dksJ t||f|
d}||||}||| }||k sJ t|||}
t
|
dksJ t|||f|
d}|||||}||| | }||k sJ t||||}
t
|
dksJ t||||f|
d}||||||	}||| | |	 }||k sJ t|ddg}
t
|
dks-J t|f|
d}|||}|t |}||k sLJ tt	||g d}
t
|
dks_J t||f|
d}||||}|t t d||g d}||k sJ tt	||d}
t
|
dksJ t||f|
d}||||}|t d||}||k sJ W d    d S 1 sw   Y  d S )Nr   r<   r   r>   r?   r@   r   r=   r^   r   r   r      z'tensorflow.linalg.einsum("ab,cd", M, N)r9   zij,klr   z1tensorflow.math.add(tensorflow.math.add(M, N), P)zItensorflow.math.add(tensorflow.math.add(tensorflow.math.add(M, N), P), Q)ztensorflow.transpose(M, [1, 0]))r   r   r=   r   zKtensorflow.transpose(tensorflow.linalg.einsum("ab,cd", M, N), [1, 2, 3, 0])zab,cd)r   r   z,tensorflow.linalg.einsum("ab,bc->acb", M, N)z
ab,bc->acb)rh   r7   ri   rj   rk   rl   rm   r2   r   r   r4   r6   rn   einsumrw   r   r   	transposer   )rW   r   r<   r>   r?   r@   r   r   mcmdr   r|   r   r   rF   rF   rG   test_codegen_extraz  s   


 
$r   c                  C   sx   t ddd} t ddd}t ddd}t| d dksJ td| d  dks(J |d || | }t|d	ks:J d S )
NAr   r=   BC)r   r   zA[0, 0]z	3*A[0, 0]z&(tensorflow.math.add((-1)*B, A))[0, 0])r2   r4   ro   )r   r   r   FrF   rF   rG   test_MatrixElement_printing  s   r   c                  C   s"   t ttt} t| dksJ d S )Nz2tensorflow.gradients(tensorflow.math.sin(x), x)[0])r   r   r   r4   r   rF   rF   rG   test_tensorflow_Derivative  s   r   c                  C   sX  t std td} tdt| fd}t|}d}t||ks)J d| d| tddftd	d
ftdd
fdfD ]\}}t| |ddt |gd}||k	 
 sWJ q;td} tdt| fd}t|}d}t||ks{J d| d| td	dftddftdd
fdfD ]\}}t| |ddt |gd}||k	 
 sJ qd S )Nr   r   g        )      ?Tz5tensorflow.where(tensorflow.math.is_nan(x), 0.0, 1.0)zIncorrect printed result z, expected naninfr   z-inf)r   r   r9   )modules)r   z5tensorflow.where(tensorflow.math.is_inf(x), 0.0, 1.0))rh   r7   r   r   r)   r4   floatr6   r   numpyrw   r*   )r   
expressionprinted_codeexpected_printed_code_input	_expected_outputrF   rF   rG   test_tensorflow_isnan_isinf  s&    , ,r   )F)fr   sympy.core.functionr   sympy.core.symbolr   sympyr   0sympy.tensor.array.expressions.array_expressionsr   r   r   r   sympy.core.relationalr	   r
   r   r   r   r   sympy.externalr   sympy.functionsr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   sympy.codegen.cfunctionsr)   r*   sympy.matricesr+   r,   r-   r.   sympy.matrices.expressionsr/   r0   r1   r2   r3   sympy.printing.tensorflowr4   3sympy.tensor.array.expressions.from_matrix_to_arrayr5   sympy.utilities.lambdifyr6   sympy.testing.pytestr7   r8   rh   r9   osenvironr<   r>   r?   r@   r   r   r   r   rB   llor   m3x3	m3x3sympyr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rF   rF   rF   rG   <module>   s`     l



 


`
5J