
    ij                        d dl Z d dlZd dlZd dlmc mZ 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 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 d dlmZ d ZddZ ej8                  dd      d        Zd Zd Zd Z d Z!d Z"d Z#ddZ$d Z% ej8                  dd      d        Z&d Z'd Z(ddZ)ddZ*ddZ+dd Z, ejZ                  d!      d"        Z.d# Z/d$ Z0dd%Z1d& Z2dd'Z3dd(Z4dd)Z5dd*Z6dd+Z7ejp                  d,        Z9ejp                  d-        Z: ejZ                  d!      d.        Z; ejZ                  d!      d/        Z< ejZ                  d!      d0        Z=d1 Z> ejZ                  d!      d2        Z?dd3Z@dd4ZAdd5ZBdd6ZCdd7ZDdd8ZEd9 ZFd: ZGd; ZHd< ZId= ZJd> ZKd? ZLd@ ZMdA ZNdB ZOdC ZPdD ZQ ejZ                  d!      dE        ZRdF ZSddGZT ejZ                  d!      dH        ZU ejZ                  d!      dI        ZV ejZ                  d!      dJ        ZWejp                  dK        ZXejp                  dL        ZYddMZZddNZ[ddOZ\ddPZ]dQ Z^dR Z_ddSZ`ddTZaddUZbddVZc ejZ                  d!      dW        ZddX ZedY ZfddZZgdd[Zhd\ Ziejp                  d]        Zjejp                  d^        Zkd_ Zl ejZ                  d!      d`        ZmddaZndb Zodc Zp ejZ                  d!      dd        ZqddeZrddfZsdg ZtddhZudi Zvdj Zwdk Zxdl ZyddmZz ejZ                  d!      dn        Z{do Z|ddpZ}ddqZ~ejp                  dr        ZddsZ ejZ                  d!      dt        Z ejZ                  d!      du        Zdv Zdw Zdx Zdy Zdz Zd{ Zd| Zd} Z	 dd~Zejp                  d        Zejp                  d        Z ejZ                  d!      d        Zejp                  d        Zd Zd Zd Zd Zd ZddZ ej8                  dd      d        Zd ZddZdddZddZ ej8                  dd      d        Zd Zd Zd Zd ZddZddZddZddZddZddZddZddZd dZddZd dZddZddZddZddZd Zd Zd ZddZddZd ZddZd dZddZddZd dZd Zd Z ejZ                  d!      d        Zejp                  d        ZddZd ZddZddZ ejZ                  d!      d        Z ejZ                  d!      d        Z ejZ                  d!      d        Zd Z ejZ                  d!      d        Zd ZddZddZddZddZddZd ZddÄZddĄZ ejZ                  d!      dń        Z ejZ                  d!      dƄ        ZѐddǄZ ejZ                  d!      ddȄ       ZdɄ ZddʄZՐdd˄Zdd̄Zdd̈́Zd΄ Zdτ ZdЄ Zdф Zd҄ ZdddӜdԄZddՄZej                  dք        Zdׄ Zd؄ Zdل Z ejZ                  d!      dڄ        Zdۄ Z ejZ                  d!      d܄        Z ejZ                  d!      d݄        ZddބZdd߄ZddZddZddZddZddZd Zd Zd ZddZddZd ZddZddZ	 	 	 	 	 	 	 d	dZd Zy(
      N)export)config)dtypes)canonicalize_axis)to_tuple_or_list)standardize_dtype)nnsparse)cast)convert_to_tensorc                     t        | d      r6| j                  }t        |t        j                        sy|j
                  dk(  S t        j                         dk(  S )NdeviceFcpu)hasattrr   
isinstancejaxDeviceplatformdefault_backend)xr   s     p/var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/keras/src/backend/jax/numpy.py	_uses_cpur      sK    q(&#**-%'' ""$--    c                     | j                   dk  rt        d| j                          t        |      dk7  s|d   |d   k(  rt        d| d      t        j                  | ||      S )z5Rotate an array by 90 degrees in the specified plane.   zBInput array must have at least 2 dimensions. Received: array.ndim=r      zInvalid axes: z3. Axes must be a tuple of two different dimensions.)kaxes)ndim
ValueErrorlenjnprot90)arrayr   r   s      r   r$   r$       s|    zzA~$$)JJ<1
 	
 4yA~aDG+TF #( (
 	
 99Uad++r   T)linearuse_sparsifyc                 Z    t        |       } t        |      }t        j                  | |      S N)r   r#   addx1x2s     r   r*   r*   /   s&    	2	B	2	B772r?r   c                 z    t        |       } t        t        j                  |       t	        j
                               S r)   )r   r   r#   bartlettr   floatxr   s    r   r/   r/   6   '    !AQ11r   c                 z    t        |       } t        t        j                  |       t	        j
                               S r)   )r   r   r#   hammingr   r0   r1   s    r   r4   r4   ;   '    !AA00r   c                 z    t        |       } t        t        j                  |       t	        j
                               S r)   )r   r   r#   hanningr   r0   r1   s    r   r7   r7   @   r5   r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   	heavisider+   s     r   r9   r9   E   '    	2	B	2	B==R  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   hypotr+   s     r   r<   r<   K   '    	2	B	2	B99Rr   c                 |    t        |       } t        t        j                  | |      t	        j
                               S r)   )r   r   r#   kaiserr   r0   )r   betas     r   r?   r?   Q   s*    !A

1d#V]]_55r   Fc           	         |st        | t        j                        rSt        | t        j                        r|t        |t        j                        st        d      | j                  |j                  urHt        j                  t        j                  | j                  |j                              st        d      |j                  }| j                  } t        | j                        dkD  rdnd}t        j                  t        j                  |       dz         }t        j                  | |d      }|t        j                  ||dz         }||z  }t        j                   ||f      }|S t        | j                        dk(  rV|fd	}	t#        t%        |	|             }
n$fd
}	t#        t%        |	t'        | |                  }
t        j(                  |
      S t        j*                  | |      S )Nz$`x` and `weights` must both be BCOOsz2`x` and `weights` BCOOs must have the same indicesr   r   Tr
   r   r   c                 2    t        j                  |       S )N)	minlengthr#   bincount)arrrD   s    r   bincount_fnzbincount.<locals>.bincount_fnu   s    ||C9==r   c                 @    t        j                  | d   | d         S )Nr   r   weightsrD   rE   )arr_wrD   s    r   rH   zbincount.<locals>.bincount_fn{   s#    ||!HeAh) r   rJ   )r   
jax_sparseBCOOr!   indicesr#   allequaldatar"   shapemaximummaxr	   one_hotexpand_dimsbcoo_reduce_sumlistmapzipstackrF   )r   rK   rD   r   reduction_axis	maxlengthone_hot_encodingexpanded_weightsoutputsrH   	bincountss     `        r   rF   rF   V   s    Az/a)"!':??;$%KLL99GOO377399QYY#HI(P  ",,A!!''lQ.AKK
Q	:	::a4@"w8JK/2BB,, "
 
177|q?> Sa01I
 Sc!Wo>?Iyy##<<7i@@r   c                     |D cg c]  }t        |       }}t        t        d |D                    }t        |      dk(  r|d   dk(  rd}nd }||d<   t	        j
                  | g|i |S c c}w )Nc              3   F   K   | ]  }t        |j                          y wr)   )r   dtype.0r   s     r   	<genexpr>zeinsum.<locals>.<genexpr>   s     CQ'0Cs   !r   r   int8int32preferred_element_type)r   rY   setr"   r#   einsum)
subscriptsoperandskwargsr   r   rk   s         r   rm   rm      s~    .67!!$7H7 #C(CCDF
6{aF1I/!(!%'=F#$::j686v66 8s   A-c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   subtractr+   s     r   rr   rr      s'    	2	B	2	B<<Br   c                     t        |       } t        |      }t        | j                        }t        |j                        }|dk(  r|dk(  rd}nd }t        | t        j
                        st        |t        j
                        rt        t        d      s,t	        j                  t        j                        t        _
        t        | t        j                        r.t	        j                  | t        | j                        dz
  d      } t        |t        j                        r.t	        j                  |t        |j                        dz
  d      }t        j                  | ||      S t        j                  | ||      S )Nri   rj   sparse_matmulr   warn)n_batchon_inefficient)rk   )r   r   re   r   rM   	JAXSparser   matmulsparsifyr#   rt   rN   bcoo_update_layoutr"   rS   )r,   r-   x1_dtypex2_dtyperk   s        r   ry   ry      s2   	2	B	2	B !*H *H6h&0!(!%"j**+z
J  0 v/#-#6#6szz#BF b*//*..CMA-fB b*//*..CMA-fB ##+A $ 
 	
 ::b"5KLLr   c                    t        |       } t        |      }t        | t        j                        r.t        |t        j                        r| j                  |j                  u r| j
                  s*t        j                  |       } t        j                  |      }t        j                  t        j                  | j                  |j                        | j                  f| j                  dd      S t        j                  | |      S t        j                  | |      }t        j                  || j                  f| j                  | j                  | j
                        S t        |t        j                        rYt        j                  ||       }t        j                  ||j                  f|j                  |j                  |j
                        S t        j                  | |      S )NT)rS   indices_sortedunique_indices)r   r   rM   rN   rO   r   bcoo_sum_duplicatesr#   multiplyrR   rS   bcoo_multiply_sparsebcoo_multiply_denser   )r,   r-   out_datas      r   r   r      st   	2	B	2	B"joo&b*//*zzRZZ' ((#77;B#77;B!\\"''2773RZZ@((#'#'	  "66r2>> "55b"=H??2::&hh!00!00	  
B
	(11"b9rzz"((,,,,	
 	
 <<Br   c                     t                t         j                        }t        j                   j                  d      }d|v s|dk(  r|}n|}t         t        j                        r|'t        t        t         j                                    }t        j                  | j                  d      \  }}}t        j                   fd|D              }	t        j                    |      }
t        j                  |
j"                  j%                  |      |	z  |
j&                  f|
j                        }
|rt        j(                  |
||	      }
|
S t+        j,                   |||
      }
t/        |
|      S )Nfloat32intboolFinsert_dimsc              3   <   K   | ]  }j                   |     y wr)   rS   )rg   ir   s     r   rh   zmean.<locals>.<genexpr>   s     ?1AGGAJ?s   rB   r   rS   broadcast_dimensionsaxiskeepdimsre   )r   r   re   r   result_typer   rM   rN   tupleranger"   rS   r   $axis_shape_dims_for_broadcast_in_dimmathprodrX   rR   astyperO   bcoo_broadcast_in_dimr#   meanr   )r   r   r   	ori_dtypecompute_dtyperesult_dtypecanonical_axiskeep_dims_shaper   divisoroutputs   `          r   r   r      sD   !A!!''*I &&qww	:M	Y&0$ !Z__%<s177|,-D
 77!''u
		
  ))???++ANC[[-7H,,
   55%%9F
 !$OFL))r   c                 J    t        |       } t        j                  | |||      S N)r   r   initial)r   r#   rU   r   r   r   r   s       r   rU   rU   
  !    !A7714(GDDr   c                 `    |xs t        j                         }t        j                  | |      S N)re   )r   r0   r#   onesrS   re   s     r   r   r     s#    $V]]_E88E''r   c                 `    |xs t        j                         }t        j                  | |      S r   )r   r0   r#   zerosr   s     r   r   r     #    $V]]_E99U%((r   )r&   c                 B    t        |       } t        j                  |       S r)   )r   r#   absoluter1   s    r   r   r         !A<<?r   c                     t        |       S r)   )r   r1   s    r   absr     s    A;r   c                 B    t        |       } t        j                  |       S r)   )r   r#   fabsr1   s    r   r   r   #      !A88A;r   c                 2    t        j                  | ||      S Nr   r   )r#   rP   r   r   r   s      r   rP   rP   (      7714(33r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S Nint64)r   r   re   r   r0   r   r   floatr   r#   angler   re   s     r   r   r   ,  sW    !A!W,""177E2QA99Q<r   c                 2    t        j                  | ||      S r   )r#   anyr   s      r   r   r   6  r   r   c                 2    t        j                  | ||      S r   )r#   amaxr   s      r   r   r   :      88AD844r   c                 2    t        j                  | ||      S r   )r#   aminr   s      r   r   r   >  r   r   c                 ^    t        |       } t        |      }t        j                  | ||      S Nr   )r   r#   append)r,   r-   r   s      r   r   r   B  s)    	2	B	2	B::b"4((r   c                     d }|M ||       g}||j                   ||             ||j                   ||             t        j                  | }t        |      }t	        j
                  | |||      S )Nc                 ~    t        | d      r| j                  S t        j                  |       rt        S t        |       S )Nre   )r   re   
jax_exportis_symbolic_dimr   typer1   s    r   	get_dtypezarange.<locals>.get_dtypeI  s1    1g77N%%a(JAwr   )stepre   )r   r   r   r   r#   arange)startstopr   re   r   dtypes_to_resolves         r   r   r   H  s|     }&u-.$$Yt_5$$Yt_5""$56e$E::eTE::r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   arccosr   s     r   r   r   [  W    !A!W,""177E2QA::a=r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   arccoshr   s     r   r   r   f  W    !A!W,""177E2QA;;q>r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   arcsinr   s     r   r   r   q  r   r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   arcsinhr   s     r   r   r   |  r   r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   arctanr   s     r   r   r     r   r   c                     t        |       } t        |      }t        j                  | j                  |j                  t              }t        | |      } t        ||      }t        j                  | |      S r)   )r   r   r   re   r   r   r#   arctan2r,   r-   re   s      r   r   r     sY    	2	B	2	Brxx59E	b%B	b%B;;r2r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   arctanhr   s     r   r   r     r   r   c                    t        |       } t        | j                        }d|vs| j                  dk(  st	        |       st        j                  | ||      S t        j                  |d      }t        | |      } | dk(  t        j                  |       z  }t        j                  |t        j                  | j                        j                   |       } t        j                  | ||      S Nr   r   r   r           )r   r   re   r    r   r#   argmaxr   r   r   signbitwherefinfotinyr   r   r   re   is_negative_zeros        r   r   r         !Aagg&Eeqvv{)A,zz!$:: ui0EQASCKKN2		"SYYqww%7%<%<$<a@A::adX66r   c                    t        |       } t        | j                        }d|vs| j                  dk(  st	        |       st        j                  | ||      S t        j                  |d      }t        | |      } | dk(  t        j                  |       z  }t        j                  |t        j                  | j                        j                   |       } t        j                  | ||      S r   )r   r   re   r    r   r#   argminr   r   r   r   r   r   r   r   s        r   r   r     r   r   c                     t        |       } | j                  dk(  rt        j                  | d       S t        j                  | |      S )Nr   r   )r   r    r#   argsortr   r   s     r   r   r     s9    !Avv{{{14((;;qt$$r   c                 0    t        j                  | |      S r   )r#   r%   r   s     r   r%   r%     s    99Qe$$r   c                 <    t        |       } | j                  |      S r   )r   viewr   s     r   r   r     s    !A666r   c                    t        |       } | j                  t        g}|&t        |      }|j                  |j                         t	        j
                  | }t        | |      } |t        ||      }t        j                  | ||      S )N)rK   r   )	r   re   r   r   r   r   r   r#   average)r   r   rK   r   re   s        r   r   r     s~    !A%(#G,  / 12EQAw&;;q'55r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   bitwise_andr   ys     r   r   r     '    !A!A??1a  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   invertr1   s    r   bitwise_invertr        !A::a=r   c                     t        |       S r)   )r  r1   s    r   bitwise_notr	    s    !r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   
bitwise_orr  s     r   r  r    s'    !A!A>>!Qr   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   bitwise_xorr  s     r   r  r    r  r   c                 z    t        |       } t        |t              st        |      }t        j                  | |      S r)   )r   r   r   r#   
left_shiftr  s     r   bitwise_left_shiftr    s1    !Aaa >>!Qr   c                     t        | |      S r)   )r  r  s     r   r  r    s    a##r   c                 z    t        |       } t        |t              st        |      }t        j                  | |      S r)   )r   r   r   r#   right_shiftr  s     r   bitwise_right_shiftr    s1    !Aaa ??1a  r   c                     t        | |      S r)   )r  r  s     r   r  r    s    q!$$r   c                 z    t        |       } t        t        j                  |       t	        j
                               S r)   )r   r   r#   blackmanr   r0   r1   s    r   r  r    r2   r   c                 D    t        |       } t        j                  | |      S r)   )r   r#   broadcast_to)r   rS   s     r   r  r    s    !AAu%%r   c                 B    t        |       } t        j                  |       S r)   )r   r#   cbrtr1   s    r   r  r    r   r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   ceilr   s     r   r  r  !  W    !A!W,""177E2QA88A;r   c                     t        |       } t        | j                        dk(  rt        | d      } t	        j
                  | ||      S )Nr   rj   )r   r   re   r   r#   clip)r   x_minx_maxs      r   r   r   ,  s<    !A!V+G88Aue$$r   c                    t        j                  d | D              }|t        |       k(  r9t        |t        | d   j                              }t        j                  | |      S |rC| D cg c]7  }t        |t
        j                        r|j                         n
t        |      9 } }n| D cg c]  }t        |       } }t        j                  | |      S c c}w c c}w )Nc              3   P   K   | ]  }t        |t        j                           y wr)   )r   rM   rN   rf   s     r   rh   zconcatenate.<locals>.<genexpr>4  s     IjJOO<Is   $&r   )	dimensionr   )builtinssumr"   r   rS   rM   bcoo_concatenater   rx   todenser   r#   concatenate)xsr   
bcoo_countr   s       r   r*  r*  3  s    IbIIJSW s2a5;;'78**2>>	
 	
  !Z112 IIK"1%&
 
 -//q"//??2D))
 0s   *<C-Cc                 B    t        |       } t        j                  |       S r)   r   r#   	conjugater1   s    r   r/  r/  D      !A==r   c                 B    t        |       } t        j                  |       S r)   r.  r1   s    r   conjr2  J  r0  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   copyr1   s    r   r4  r4  P      !A88A;r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   cosr   s     r   r7  r7  V  W    !A!W,""177E2QA771:r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   coshr   s     r   r:  r:  a  r  r   c                 D    t        t        j                  | |      d      S )Nr   rj   )r   r#   count_nonzeror   s     r   r<  r<  l  s    !!!$/99r   c                 d    t        |       } t        |      }t        j                  | |||||      S )N)axisaaxisbaxiscr   )r   r#   cross)r,   r-   r>  r?  r@  r   s         r   rA  rA  p  s9    	2	B	2	B99

 r   c                 H    t        |       } t        j                  | ||      S N)r   re   )r   r#   cumprodr   r   re   s      r   rD  rD  }  s    !A;;qt511r   c                 H    t        |       } t        j                  | ||      S rC  )r   r#   cumsumrE  s      r   rG  rG    s    !A::ad%00r   c                 B    t        |       } t        j                  |       S r)   )r   r#   deg2radr1   s    r   rI  rI        !A;;q>r   c                 B    t        |       } t        j                  |       S r)   )r   r#   rad2degr1   s    r   rL  rL    rJ  r   c                 F    t        |       } t        j                  | |      S N)r   )r   r#   diagr   r   s     r   rO  rO        !A88Ar   c                 F    t        |       } t        j                  | |      S rN  )r   r#   diagflatrP  s     r   rS  rS    s    !A<<Qr   c                 J    t        |       } t        j                  | |||      S N)offsetaxis1axis2)r   r#   diagonalr   rV  rW  rX  s       r   rY  rY    s)    !A<<		 r   c                 H    t        |       } t        j                  | ||      S )N)nr   )r   r#   diff)ar\  r   s      r   r]  r]    s    !A88A&&r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   digitize)r   binss     r   r`  r`    s'    !AT"D<<4  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   dotr+   s     r   rc  rc    &    	2	B	2	B772r?r   c                 ,    t        j                  |       S r)   )r#   dstackr+  s    r   rf  rf        ::b>r   c                 `    |xs t        j                         }t        j                  | |      S r   )r   r0   r#   emptyr   s     r   rj  rj    r   r   c                 0    t        j                  | |      S r   )r#   
empty_liker   s     r   rl  rl        >>!5))r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   rQ   r+   s     r   rQ   rQ     r=   r   c                     t        |       } t        | j                        }d|v s|dk(  rt        | t	        j
                               } t        j                  |       S Nr   r   )r   r   re   r   r   r0   r#   expr   r   s     r   rq  rq    sI    !A!!''*I	Y&0FMMO$771:r   c                     t        |       } t        | j                        }d|v s|dk(  rt        | t	        j
                               } t        j                  |       S rp  )r   r   re   r   r   r0   r#   exp2rr  s     r   rt  rt    sI    !A!!''*I	Y&0FMMO$88A;r   c                     t        |       } t        | t        j                        r>t	        j
                  || j                  d      \  }}}t        j                  | ||      S t        j                  | |      S )NTr   r   )
r   r   rM   rN   r   r   rS   r   r#   rW   )r   r   _result_shaper   s        r   rW   rW     ss    !A!Z__%
 77!''t
		
  //\8L
 	
 ??1d##r   c                     t        |       } t        | j                        }d|v s|dk(  rt        | t	        j
                               } t        j                  |       S rp  )r   r   re   r   r   r0   r#   expm1rr  s     r   ry  ry    sI    !A!!''*I	Y&0FMMO$99Q<r   c                 0    t        j                  | |      S r   )r#   flipr   s     r   r{  r{    s    88AD!!r   c                 B    t        |       } t        j                  |       S r)   )r   r#   fliplrr1   s    r   r}  r}    r  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   flipudr1   s    r   r  r    r  r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   floorr   s     r   r  r    sW    !A!W,""177E2QA99Q<r   c                 b    |xs t        j                         }t        j                  | ||      S r   )r   r0   r#   full)rS   
fill_valuere   s      r   r  r    s%    $V]]_E88E:U33r   c                 2    t        j                  | ||      S r   )r#   	full_like)r   r  re   s      r   r  r    s    ==Je44r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   gcdr+   s     r   r  r    rd  r   c                 8    t        j                  | |||||      S )N)numendpointre   r   )r#   	geomspace)r   r   r  r  re   r   s         r   r  r    s     ==txu4 r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   greaterr+   s     r   r  r  #  s'    	2	B	2	B;;r2r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   greater_equalr+   s     r   r  r  )  s)    	2	B	2	BR$$r   c                 ,    t        j                  |       S r)   )r#   hstackrg  s    r   r  r  /  rh  r   c                 D    t        |       } t        j                  | |      S r)   )r   r#   hsplitr   indices_or_sectionss     r   r  r  3      !A::a,--r   c                 `    |xs t        j                         }t        j                  | |      S r   )r   r0   r#   identity)r\  re   s     r   r  r  8  s#    $V]]_E<<''r   c                 B    t        |       } t        j                  |       S r)   )r   r#   imagr1   s    r   r  r  =  r5  r   c                     t        |       } t        | j                        }|dv rd}n|dvrt        j                         }t        | |      } t        j                  |       S )N)r   float64r  )bfloat16float16)r   r   re   r   r0   r   r#   i0r   s     r   r  r  C  sU    !Aagg&E$$	-	-QA66!9r   c                 `    t        |       } t        |      }t        j                  | ||||      S r)   )r   r#   iscloser,   r-   rtolatol	equal_nans        r   r  r  N  s-    	2	B	2	B;;r2tT955r   c                 `    t        |       } t        |      }t        j                  | ||||      S r)   )r   r#   allcloser  s        r   r  r  T  s-    	2	B	2	B<<BdI66r   c                 B    t        |       } t        j                  |       S r)   )r   r#   isfiniter1   s    r   r  r  Z  r   r   c                 `    t        |       } t        |      }t        j                  | |||      S )N)assume_uniquer  )r   r#   isin)r,   r-   r  r  s       r   r  r  `  s+    	2	B	2	B88B-GGr   c                 B    t        |       } t        j                  |       S r)   )r   r#   isinfr1   s    r   r  r  f      !A99Q<r   c                 B    t        |       } t        j                  |       S r)   )r   r#   isnanr1   s    r   r  r  l  r  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   isneginfr1   s    r   r  r  r      !A<<?r   c                 B    t        |       } t        j                  |       S r)   )r   r#   isposinfr1   s    r   r  r  w  r  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   isrealr1   s    r   r  r  |  r  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   kronr+   s     r   r  r    '    	2	B	2	B88Br   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   lcmr+   s     r   r  r    rd  r   c                     t        |       } t        |      }t        |j                        t        j                  vrt        d|j                         t        j                  | |      S )Nz;ldexp exponent must be an integer type. Received: x2 dtype=)r   r   re   r   	INT_TYPES	TypeErrorr#   ldexpr+   s     r   r  r    s`    	2	B	2	B"&*:*::""$((-
 	

 99Rr   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   lessr+   s     r   r  r    r  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   
less_equalr+   s     r   r  r    '    	2	B	2	B>>"b!!r   c           	      :    t        j                  | ||||||      S )N)r  r  retstepre   r   )r#   linspace)r   r   r  r  r  re   r   s          r   r  r    s*     << r   c                     t        |       } t        | j                        dk(  rt        | t	        j
                               } t        j                  |       S r   )r   r   re   r   r   r0   r#   logr1   s    r   r  r    s=    !A!W,FMMO$771:r   c                     t        |       } t        | j                        dk(  rt        | t	        j
                               } t        j                  |       S r   )r   r   re   r   r   r0   r#   log10r1   s    r   r  r    =    !A!W,FMMO$99Q<r   c                     t        |       } t        | j                        dk(  rt        | t	        j
                               } t        j                  |       S r   )r   r   re   r   r   r0   r#   log1pr1   s    r   r  r    r  r   c                     t        |       } t        | j                        dk(  rt        | t	        j
                               } t        j                  |       S r   )r   r   re   r   r   r0   r#   log2r1   s    r   r  r    =    !A!W,FMMO$88A;r   c                     t        |       } t        |      }t        j                  | j                  |j                  t              }t        | |      } t        ||      }t        j                  | |      S r)   )r   r   r   re   r   r   r#   	logaddexpr   s      r   r  r    sY    	2	B	2	Brxx59E	b%B	b%B==R  r   c                     t        |       } t        |      }t        j                  | j                  |j                  t              }t        | |      } t        ||      }t        j                  | |      S r)   )r   r   r   re   r   r   r#   
logaddexp2r   s      r   r  r    sY    	2	B	2	Brxx59E	b%B	b%B>>"b!!r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   logical_andr+   s     r   r  r    '    	2	B	2	B??2r""r   c                 B    t        |       } t        j                  |       S r)   )r   r#   logical_notr1   s    r   r  r    s    !A??1r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   
logical_orr+   s     r   r  r    r  r   c           	      :    t        j                  | ||||||      S )N)r  r  basere   r   )r#   logspace)r   r   r  r  r  re   r   s          r   r  r    s(    << r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   rT   r+   s     r   rT   rT     '    	2	B	2	B;;r2r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   fmaxr+   s     r   r  r  
  r  r   c                    t        |t              rt        |      }t        |       } t	        | j
                        dk(  rt        | t        j                               } t        j                  | ||      }|du rL|J|j                  | j                  k  r1t        j                  |d      }|j                  | j                  k  r1|S )Nr   r   Tr   )r   rY   r   r   r   re   r   r   r0   r#   medianr    rW   )r   r   r   results       r   r  r    s    $T{!A!W,FMMO$ZZx8F 4DLkkAFF"__V"5F kkAFF"Mr   xy)indexingc                 ,    t        j                  |d| iS )Nr  )r#   meshgrid)r  r   s     r   r  r  !  s    <<.X..r   c                 J    t        |       } t        j                  | |||      S r   )r   r#   minr   s       r   r  r  %  r   r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   minimumr+   s     r   r  r  *  r  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   fminr+   s     r   r  r  1  r  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   modr+   s     r   r  r  7  rd  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   fmodr+   s     r   r  r  =  r  r   c                 2    t        j                  | ||      S )N)sourcedestination)r#   moveaxis)r   r  r  s      r   r  r  C  s    <<&kBBr   c                 H    t        |       } t        j                  | ||      S r   )r   r#   	nanargmaxr   s      r   r  r  G      !A==99r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   	nanargminr   s      r   r  r  L  r  r   c                 H    t        |       } t        j                  | ||      S rC  )r   r#   	nancumsumrE  s      r   r  r  Q  s    !A==U33r   c                 H    t        |       } t        j                  | ||      S rC  )r   r#   
nancumprodrE  s      r   r  r  V  s    !A>>!$e44r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   nanmaxr   s      r   r  r  [      !A::adX66r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   nanmeanr   s      r   r
  r
  `      !A;;qth77r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   	nanmedianr   s      r   r  r  e  r  r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   nanminr   s      r   r  r  j  r  r   c                 b    t        |       } t        |      }t        j                  | ||||      S Nr   methodr   )r   r#   nanpercentiler   qr   r  r   s        r   r  r  o  s/    !A!AQVhOOr   c                 H    t        |       } t        j                  | ||      S r   )r   r#   nanprodr   s      r   r  r  u  r  r   c                 b    t        |       } t        |      }t        j                  | ||||      S r  )r   r#   nanquantiler  s        r   r  r  z  s-    !A!A??1ad6HMMr   c                 H    t        |       } t        j                  | ||      S r   )r   r#   nanstdr   s      r   r  r    r  r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   nansumr   s      r   r  r    r  r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   nanvarr   s      r   r   r     r  r   c                 J    t        |       } t        j                  | |||      S )N)nanposinfneginf)r   r#   
nan_to_num)r   r"  r#  r$  s       r   r%  r%    s!    !A>>!VFCCr   c                 ,    t        j                  |       S r)   )r#   r    r1   s    r   r    r          88A;r   c                 ,    t        j                  |       S r)   )r#   nonzeror1   s    r   r)  r)    s    ;;q>r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   	not_equalr+   s     r   r+  r+    r:   r   c                 0    t        j                  | |      S r   )r#   	ones_liker   s     r   r-  r-    s    ==%((r   c                 0    t        j                  | |      S r   )r#   
zeros_liker   s     r   r/  r/    rm  r   c                 .    t        j                  | |      S r)   )r#   outerr+   s     r   r1  r1    s    99Rr   c                     t        |       } i }||dk7  rt        d|       ||d<   t        j                  | |fd|i|S )NconstantzZArgument `constant_values` can only be provided when `mode == 'constant'`. Received: mode=constant_valuesmode)r   r!   r#   pad)r   	pad_widthr5  r4  rp   s        r   r6  r6    sc    !AF":""&) 
 %4 !771i5d5f55r   c                 b    t        |       } t        |      }t        j                  | ||||      S r  )r   r#   
percentiler  s        r   r9  r9    s-    !A!A>>!QT&8LLr   c                 J    t        |       } t        j                  | |||      S )Nr   )r   r#   r   )r   r   r   re   s       r   r   r     s!    !A88AD85AAr   c                 H    t        |       } t        j                  | ||      S r   )r   r#   ptpr   s      r   r<  r<    s    !A7714(33r   c                    t        |       } t        |      }t        | j                        dk(  rt        | t	        j
                               } t        j                  | ||||      }|du ra|_| j                  t        |j                        dkD  rdndz   }|j                  |k  r't        j                  |d      }|j                  |k  r'|S )Nr   r  Tr   r   r  r   )r   r   re   r   r   r0   r#   quantiler    r"   rS   rW   )r   r  r   r  r   r  result_ndims          r   r>  r>    s    !A!A!W,FMMO$\\!QT&8LF 4DLffS\A%51=kkK'__V"5F kkK'Mr   c                 B    t        |       } t        j                  |       S r)   )r   r#   ravelr1   s    r   rA  rA    s    !A99Q<r   c                 D    t        |       } t        j                  | |      S r)   )r   r#   unravel_index)rO   rS   s     r   rC  rC    s    (GWe,,r   c                 B    t        |       } t        j                  |       S r)   )r   r#   realr1   s    r   rE  rE    r5  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   
reciprocalr1   s    r   rG  rG    s    !A>>!r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   repeat)r   repeatsr   s      r   rI  rI    s    !A::at,,r   c                     t        | t        j                        r@ddlm} |j                  | j                  |d      }d |vr|}t        j                  | |      S t        |       } t        j                  | |      S )Nr   )operation_utils	new_shape)	new_sizes)r   rM   rN   keras.src.opsrL  compute_reshape_output_shaperS   bcoo_reshaper   r#   reshape)r   newshaperL  output_shapes       r   rR  rR    sm    !Z__%1 'CCGGX{
 |##H&&qH==!A;;q(##r   c                 2    t        j                  | ||      S r   )r#   roll)r   shiftr   s      r   rV  rV    s    88Au4((r   c                 ~    t        |       dk7  rt        d| j                         t        j                  | ||      S )Nr   z`searchsorted` only supports 1-D sorted sequences. You can use `keras.ops.vectorized_map` to extend it to N-D sequences. Received: sorted_sequence.shape=)side)r    r!   rS   r#   searchsorted)sorted_sequencevaluesrY  s      r   rZ  rZ  
  sK    O!% &5%:%:$;=
 	
 OV$??r   c                 B    t        |       } t        j                  |       S r)   )r   r#   signr1   s    r   r^  r^    r5  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   r   r1   s    r   r   r     s    !A;;q>r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   sinr   s     r   ra  ra  !  r8  r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   sincr   s     r   rc  rc  ,  sW    !A!W,""177E2QA88A;r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   sinhr   s     r   re  re  6  r  r   c                 ,    t        j                  |       S r)   )r#   sizer1   s    r   rg  rg  A  r'  r   c                 F    t        |       } t        j                  | |      S r   )r   r#   sortr   s     r   ri  ri  E  s    !A88AD!!r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   splitr   r  r   s      r   rk  rk  J  s     !A99Q+$77r   c                 H    t        |       } t        j                  | ||      S r   )r   r#   array_splitrl  s      r   rn  rn  O  s     !A??11==r   c                 j    | D cg c]  }t        |       } }t        j                  | |      S c c}w r   )r   r#   r\   )r   r   ts      r   r\   r\   T  s1    '()!	1	)A)99QT"" 	*s   0c                     t        |       } t        | j                        dk(  rt        | t	        j
                               } t        j                  | ||      S )Nr   r   )r   r   re   r   r   r0   r#   stdr   s      r   rr  rr  Y  sB    !A!W,FMMO$7714(33r   c                 H    t        |       } t        j                  | ||      S )N)rW  rX  )r   r#   swapaxes)r   rW  rX  s      r   rt  rt  `  s    !A<<e44r   c                 b    t        |       } t        |d      }t        j                  | ||      S NFr
   r   )r   r#   taker   rO   r   s      r   rw  rw  e  s+    !A6G88AwT**r   c                 b    t        |       } t        |d      }t        j                  | ||      S rv  )r   r#   take_along_axisrx  s      r   rz  rz  k  s-    !A6Gq'55r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   tanr   s     r   r|  r|  q  r8  r   c                     t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }t        | |      } t        j                  |       S r   )r   r   re   r   r0   r   r   r   r   r#   tanhr   s     r   r~  r~  |  r  r   c                 ^    t        |       } t        |      }t        j                  | ||      S )NrB   )r   r#   	tensordot)r,   r-   r   s      r   r  r    s)    	2	B	2	B==Rd++r   c                    t        |       } t        | j                        }d|v r|dk  rt        t	        j
                  d|      t        j                               }t        | t        j                               } t        j                  | |      } t        j                  |       } t        j                  | |      } t        | |      S t        j                  | |      S )Nr   r   
   )decimals)r   r   re   r   r   powr   r0   r#   r   rounddivide)r   r  x_dtypefactors       r   r  r    s    !A  (GHqLdhhr8,fmmo>FMMO$LLF#IIaLJJq&!AwyyX..r   c                 .    t        j                  | |      S r)   )r#   tile)r   rJ  s     r   r  r    s    88Awr   c                 J    t        |       } t        j                  | |||      S rU  )r   r#   tracerZ  s       r   r  r    s!    !A99QvU%@@r   c                 d    |xs t        j                         }t        j                  | |||      S N)Mr   re   )r   r0   r#   triNr  r   re   s       r   r  r    '    $V]]_E771Qe,,r   c                 F    t        |       } t        j                  | |      S rN  )r   r#   trilrP  s     r   r  r    rQ  r   c                 F    t        |       } t        j                  | |      S rN  )r   r#   triurP  s     r   r  r    rQ  r   c                     t        |       } t        | j                        }d|v sd|k(  r| S t        j                  |       S rp  )r   r   re   r#   truncr   s     r   r  r    s:    !Aagg&E~599Q<r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   vdotr+   s     r   r  r    r  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   innerr+   s     r   r  r    r=   r   c                 ,    t        j                  |       S r)   )r#   vstackrg  s    r   r  r    rh  r   c                 D    t        |       } t        j                  | |      S r)   )r   r#   vsplitr  s     r   r  r    r  r   excluded	signaturec                J    |
t               }t        j                  | ||      S )Nr  )rl   r#   	vectorize)pyfuncr  r  s      r   r  r    s"    5==(iHHr   c                 0    t        j                  | ||      S r)   )r#   r   )	conditionr,   r-   s      r   r   r     s    99YB''r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   r  r+   s     r   r  r    s'    	2	B	2	B::b"r   c                     t        |       } t        |      }t        j                  |dk(  d|      }t        j                  |dk(  dt        j                  | |            S )Nr   r   )r   r#   r   r  )r,   r-   safe_x2s      r   divide_no_nanr    sN    	2	B	2	BiiaB'G99R1WaB!899r   c                     t        | |      S r)   )r  r+   s     r   true_divider    s    "b>r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   powerr+   s     r   r  r    r=   r   c                 B    t        |       } t        j                  |       S r)   )r   r#   negativer1   s    r   r  r    r   r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   	nextafterr+   s     r   r  r    r:   r   c                 B    t        |       } t        j                  |       S r)   )r   r#   squarer1   s    r   r  r    s    !A::a=r   c                     t        |       } t        | j                        dk(  rt        | t	        j
                               } t        j                  |       S r   )r   r   re   r   r   r0   r#   sqrtr1   s    r   r  r    r  r   c                    t        | t        j                        rI|%t        d t	        | j
                        D              }t        |      }t        j                  | |      S t        |       } t        j                  | |      S )Nc              3   2   K   | ]  \  }}|d k(  s|  yw)r   N )rg   r   ds      r   rh   zsqueeze.<locals>.<genexpr>  s     Dtq!Q!VDs   )
dimensionsr   )r   rM   rN   r   	enumeraterS   r   bcoo_squeezer   r#   squeezer   s     r   r  r    sf    !Z__%<Dy'9DDD%&&qT::!A;;qt$$r   c                 X   t        |       } t        | t        j                        rot	        | j
                        }|t        t        |      d d d         }n&g }|D ]  }t        ||      }|j                  |       ! t        j                  | |      S t        j                  | |      S )Nr  )permutationrB   )r   r   rM   rN   r"   rS   r   r   r   r   bcoo_transposer#   	transpose)r   r   num_dimsr  r^  s        r   r  r    s    !A!Z__%qww<<h" 56KK &%a2""1%& ((DD==&&r   c                 z    t        |       } |t        |      }t        |      }t        j                  | |||      S )N)dxr   )r   r#   	trapezoid)r  r   r  r   s       r   r  r  (  s:    !A}a 	2	B==A"400r   c                 H    t        |       } t        j                  | ||      S )N)r  
increasing)r   r#   vander)r   r  r  s      r   r  r  0  s    !A::a144r   c                     t        |       } t        j                  | j                  d      }t        j                  | j                  t              }t        t        j                  | |||      |      S )Nr   r   )r   r   r   re   r   r   r#   var)r   r   r   r   r   s        r   r  r  5  s\    !A &&qww	:M%%aggu5Lx}E r   c                    t        |       } t        | t        j                        r|'t	        t        t        | j                                    }t        j                  || j                  d      \  }}}t        j                  | |      }|rt        j                  |||      }|S t        j                  | ||      S )NFr   rB   r   r   )r   r   rM   rN   r   r   r"   rS   r   r   rX   r   r#   r'  )r   r   r   r   r   r   r   s          r   r'  r'  A  s    !A!Z__%<s177|,-D
 77!''u
		
  ++ANC  55%%9F
 7714(33r   c                 d    |xs t        j                         }t        j                  | |||      S r  )r   r0   r#   eyer  s       r   r  r  Z  r  r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   floor_divider+   s     r   r  r  _  s)    	2	B	2	BB##r   c                 Z    t        |       } t        |      }t        j                  | |      S r)   )r   r#   logical_xorr+   s     r   r  r  e  r  r   c                 B    t        |       } t        j                  |       S r)   )r   r#   corrcoefr1   s    r   r  r  k  r  r   c                 \    t        |       } t        |      }t        j                  | ||      S r)   )r   r#   	correlate)r,   r-   r5  s      r   r  r  p  s)    	2	B	2	B==R&&r   c                 2    t        j                  | ||      S )N)default)r#   select)condlist
choicelistr  s      r   r  r  v  s    ::h
G<<r   c                 h    t        |       } t        t        j                  j	                  |             S r)   )r   r   r#   linalgslogdetr1   s    r   r  r  z  s&    !A##A&''r   c                     |.t        j                  | d      } t        j                  | |d      S t        j                  | ||      S )Nr  r   r   )r#   rR  argpartition)r   kthr   s      r   r  r    sA    |KK5!3Q//As..r   c                 2    t        j                  | ||      S )N)ra  r   )r#   	histogram)r   ra  r   s      r   r  r    s    ==U33r   c                 >    t        j                  | ||||d|||	      S )NF)return_indexreturn_inversereturn_countsr   r  rg  sortedr  )r#   unique)r   r  r  r  r  r   rg  r  s           r   r  r    s0     ::	!%#
 
r   c                 D    t        |       } t        j                  | |      S r)   )r   r#   dsplitr  s     r   r  r    r  r   )r   )r   r   )Nr   F)NF)NFNr)   )NNNr  )NN)r   )r  r  r  N)r   r   r   )r   r  )2   TNr   )gh㈵>g:0yE>F)FF)r  TFNr   )r  Tr  Nr   )Nr&   F)r   NN)r3  N)left)r   )Nr   N)Ng      ?r  )valid)r  N)TFFFNNN)r&  r   r   jax.experimental.sparseexperimentalr   rM   	jax.numpynumpyr#   r   r   keras.src.backendr   keras.src.backend.commonr   &keras.src.backend.common.backend_utilsr   r   "keras.src.backend.common.variablesr   keras.src.backend.jaxr	   keras.src.backend.jax.corer   r   r   r$   elementwise_binary_unionr*   r/   r4   r7   r9   r<   r?   rF   rm   rr   ry   r   r   rU   r   r   elementwise_unaryr   r   r   rP   r   r   r   r   r   r   densifying_unaryr   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/  r2  r4  r7  r:  r<  rA  rD  rG  rI  rL  rO  rS  rY  r]  r`  rc  rf  rj  rl  rQ   rq  rt  rW   ry  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  r  r  rT   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/  r1  r6  r9  r   r<  r>  rA  rC  rE  rG  rI  rR  rV  rZ  r^  r   ra  rc  re  rg  ri  rk  rn  r\   rr  rt  rw  rz  r|  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r   elementwise_divisionr  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   <module>r     s(	     
 , ,  $ $ + D C @ $ ( + 8	., !  4@ A2
1
1
!6
-A`
7 !  4@  A M>& R%*PE
(
)
 ' (

4455);&     ' ( ' ( ' ( ' (77%%

6!
 ! $!%2
&

 ' (%*" & '
 & '
 & '
    :
2
1



 
'
 '! (!)
*    $  ' ("

 ' (4
5%.
(
 & '
67  
H ' (
 ' (




" IJ     ' (  !"#
"	 !  EB C"  /E
 !  EB CC:
:
4
5
7
8
:
7
P8
N7
7
7
D
!)*6MB
4
 
-
 & '
  
-
$)@ ' (
 ' (
 ' ( ' ("
8
>
#
45
+6 ' ( ' (, '/ (/  A
-


.
 #'$ I(  : & '
! ' (
 ' (%'15
	42-
$#
'=(
/4 		..r   