
    uki/                     &
   d dl mZ d dl mZ d dlmZ d dlmZ d dlm	Z	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$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZ d d	lmZmZmZmZmZ d d
lmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZmZmZmZ d dlmZmZmZmZmZmZmZmZmZ d dlmZmZmZ d dlmZmZmZmZmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z mZmZmZmZmZmZmZmZ d dl	m
Z
mZ d dlmZmZmZmZmZ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 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z? d dl@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJ d dlKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZz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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mZmZmZmZ d dlmZmZ d dlmZ d dlmZ  e        [y)    )fft)linalg)Array)isdtype)arrayasarray)ComplexWarningallcloseangleappendapply_along_axisapply_over_axesarangeargmaxargminargwherearoundarray_equalarray_equivarray_splitastype
atleast_1d
atleast_2d
atleast_3dbincountblockbroadcast_arraysbroadcast_shapesbroadcast_tocan_castchooseclipcolumn_stackcompressconcatconcatenateconvolvecopycorrcoef	correlatecovcrossdeletediagdiagflatdiag_indicesdiag_indices_fromdiagonaldiffdigitizedsplitdstackediff1dexpand_dimsextracteyefill_diagonalfinfofixflatnonzeroflipfliplrflipudfmaxfmin
frombufferfromfilefromfunctionfromiter
fromstringfrom_dlpackgcdget_printoptionsgradient	histogramhistogram_bin_edgeshistogram2dhistogramddhsplithstacki0identityiinfoindicesinsertinterpisclose	iscomplexiscomplexobjisreal	isrealobjisscalar
issubdtypeix_kronlcmloadmask_indicesmatrix_transposemeshgridmoveaxis
nan_to_num	nanargmax	nanargminnonzeropackbitspadpermute_dims	piecewiseprintoptionspromote_typesravelravel_multi_indexrepeatreshaperesizeresult_typerollrollaxisrot90roundsearchsortedselectset_printoptionssplitsqueezestackswapaxestiletrace	trapezoid	transposetritriltril_indicestril_indices_from
trim_zerostriutriu_indicestriu_indices_fromtrunc
unpackbitsunravel_indexunstackunwrapvandervsplitvstackwhere)empty
empty_likefull	full_like	geomspacelinspacelogspaceones	ones_likezeros
zeros_like)einsumeinsum_path)placeputput_along_axistaketake_along_axis)%bfloat16bool_r   cdoublecsingle
complex128	complex64complex_doublefloat16float32float4_e2m1fnfloat64float8_e3m4float8_e4m3float8_e4m3b11fnuzfloat8_e4m3fnfloat8_e4m3fnuzfloat8_e5m2float8_e5m2fnuzfloat8_e8m0fnufloat_int2int4int8int16int32int64int_singleuintuint2uint4uint8uint16uint32uint64)argpartitionargsortlexsort	partitionsortsort_complex)	dotinnermatmulmatvecouter	tensordotvecdotvecmatvdot)ndimshapesize)bartlettblackmanhamminghanningkaiser)
array_repr	array_str	charactercomplexfloatingdtypeeeuler_gammaflexiblefloatinggenericinexactinfintegeriterablenannewaxisnumberobject_pisavesavezsignedintegerunsignedinteger)__array_api_version____array_namespace_info__)c_	index_expmgridogridr_s_)
polypolyaddpolyderpolydivpolyfitpolyintpolymulpolysubpolyvalroots)!aminamaxanyallaveragecount_nonzerocumprodcumsumcumulative_prodcumulative_summaxmeanmedianmin	nancumsum
nancumprodnanmaxnanmean	nanmediannanminnanpercentilenanprodnanquantilenanstdnansumnanvar
percentileprodptpquantilestdsumvar)
intersect1disin	setdiff1dsetxor1dunion1dunique
unique_allunique_countsunique_inverseunique_values)eabsabsoluteacosacoshaddarccosarccosharcsinarcsinharctanarctan2arctanhasinasinhatanatanhatan2bitwise_andbitwise_countbitwise_invertbitwise_left_shiftbitwise_notbitwise_right_shift
bitwise_orbitwise_xorcbrtceilconj	conjugatecopysigncoscoshdeg2raddegreesdividedivmodequalexpexp2expm1fabsfloat_powerfloorfloor_dividefmodfrexpgreatergreater_equal	heavisidehypotimaginvertisfiniteisinfisnanisneginfisposinfldexp
left_shiftless
less_equalloglog10log1plog2	logaddexp
logaddexp2logical_andlogical_not
logical_orlogical_xormaximumminimummodmodfmultiplynegative	nextafter	not_equalpositivepowpowerrad2degradiansreal
reciprocal	remainderright_shiftrintsignsignbitsinsincsinhspacingsqrtsquaresubtracttantanhtrue_divide)
frompyfuncufunc)	vectorize)register_jax_array_methodsN(  	jax.numpyr   r   jax._src.basearrayr   ndarrayjax._src.dtypesr   !jax._src.numpy.array_constructorsr   r   jax._src.numpy.lax_numpyr	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   jax._src.numpy.array_creationr   r   r   r   r   r   r   r   r   r   r   jax._src.numpy.einsumr   r   jax._src.numpy.indexingr   r   r   r   r   jax._src.numpy.scalar_typesr   r   boolr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   jax._src.numpy.sortingr   r   r   r   r   r   "jax._src.numpy.tensor_contractionsr   r   r   r   r   r   r   r   r   jax._src.numpy.utilr   r   r   jax._src.numpy.window_functionsr   r   r   r   r   numpyr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   !jax._src.numpy.array_api_metadatar   r   jax._src.numpy.index_tricksr   r   r   r  r  r  jax._src.numpy.polynomialr  r  r  r  r  r	  r
  r  r  r  jax._src.numpy.reductionsr  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  r*  r+  r,  r-  r.  jax._src.numpy.setopsr/  r0  r1  r2  r3  r4  r5  r6  r7  r8  jax._src.numpy.ufuncsr9  r:  r;  r<  r=  r>  r?  r@  rA  rB  rC  rD  rE  rF  rG  rH  rI  rJ  rK  rL  rM  rN  rO  rP  rQ  rR  rS  rT  rU  rV  rW  rX  rY  rZ  r[  r\  r]  r^  r_  r`  ra  rb  rc  rd  re  rf  rg  rh  ri  rj  rk  rl  rm  rn  ro  rp  rq  rr  rs  rt  ru  rv  rw  rx  ry  rz  r{  r|  r}  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  jax._src.numpy.ufunc_apir  r  jax._src.numpy.vectorizer  jax._src.numpy.array_methodsr       M/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/numpy/__init__.py<module>r     s  $ ! & /
P P P P P P P P P P P P P P P P P P P P P P P P P P P P P P P P P P P P Pd   
 & & & & & & & & & &P 
 
 
          4 
        " " " " " " " " " " " " " " " " " "H     f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f f fP 
 < ; D C  r  