
    ij                     j   d dl 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 dd
Zd Zd Zd Zd Zd ZddZddZddZddZd Zd Zd ZddZddZd ZddZddZ ddZ!ddZ"ddZ#d Z$d  Z%d! Z&d" Z'd# Z(d$ Z)d% Z*dd&Z+dd'Z,dd(Z-dd)Z.dd*Z/dd+Z0d, Z1d- Z2d. Z3d/ Z4d0 Z5dd1Z6d2 Z7d3 Z8d4 Z9d5 Z:d6 Z;d7 Z<d8 Z=d9 Z>d: Z?d; Z@d< ZAd= ZBd> ZCd? ZDdd@ZEdA ZFdB ZGdC ZHdD ZIdE ZJddFZKddGZLddHZMddIZNdJ ZOdK ZPddLZQddMZRddNZSddOZTdP ZUdQ ZVdR ZWddSZXddTZYdU ZZdV Z[dW Z\dX Z]dY Z^ddZZ_d[ Z`d\ Zad] Zbdd^Zcdd_Zdd` ZeddaZfdb Zgdc Zhdd Zide Zjdf ZkddgZldh Zmdi ZnddjZodk ZpddlZqdm Zrdn Zsdo Ztdp Zudq Zvdr Zwds Zxdt Zydu Zzdv Z{	 ddwZ|dx Z}dy Z~dz Zd{ Zd| Zd} Zd~ Zd Zd ZddZd Zd ZddZdddZddZd Zd Zd Zd Zd ZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZd Zd Zd ZddZddZd ZddZddZddZddZddZd Zd Zd Zd ZddZd ZddZddZd Zd Zd Zd Zd Zd ZddZddZddZddZddZd ZddZddZd Zd ZddZddZd ZddÄZddĄZddńZddƄZdǄ ZdȄ ZdɄ Zdʄ Zd˄ Zddd̜d̈́Zdd΄Zdτ ZdЄ Zdф Zd҄ Zdӄ ZdԄ ZdՄ Zdք ZddׄZdd؄ZddلZddڄZddۄZdd܄Zdd݄Zdބ Zd߄ Zd Zd dZddZd ZddZddZ	 	 	 	 	 	 	 ddZd Zy(      N)backend)tree)config)standardize_dtype)dtypes)canonicalize_axis)standardize_axis_for_numpyconvert_to_tensorc                     | 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nprot90)arrayr   r   s      r/var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/keras/src/backend/numpy/numpy.pyr   r      s|    zzA~$$)JJ<1
 	
 4yA~aDG+TF #+ +
 	
 88EQT**    c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S Ndtype)

isinstanceintfloatr   r   result_typegetattrtyper   addx1x2r   s      r   r"   r"      s    b3,'r"b3,'r"GT"X&GT"X&E 
2u	%B	2u	%B66"b>r   c                 \   t        j                  t        |      }t        t	        d |D                    }t        |      dk(  r|d   dk(  rdd}nt        j                  | }|dk(  rdt        j                  fd|      }t        j                  | g|i |j                  |      S )	Nc              3   F   K   | ]  }t        |j                          y wN)r   r   ).0xs     r   	<genexpr>zeinsum.<locals>.<genexpr>,   s      N!2177!; Ns   !r   r   int8int32bfloat16float32c                 &    | j                        S r(   astype)r*   compute_dtypes    r   <lambda>zeinsum.<locals>.<lambda>8   s    AHH],C r   )r   map_structurer   listsetr   r   r   r   einsumr2   )
subscriptsoperandskwargsdtypes_to_resolveresult_dtyper3   s        @r   r8   r8   *   s    !!"3X>HS NX NNO "'8';v'E))+<=$J&%M!!"CXNH99Z5(5f5<<\JJr   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   subtractr#   s      r   r?   r?   <       b3,'r"b3,'r"GT"X&GT"X&E 
2u	%B	2u	%B;;r2r   c                 ~   t        |       } t        |      }t        | j                        }t        |j                        }|dk(  r|dk(  rd}n*t        j                  | j                  |j                        }| j                  |      } |j                  |      }t        j                  | |      j                  |      S )Nr,   r-   )r   r   r   r   r   r2   r   matmul)r$   r%   x1_dtypex2_dtyper   s        r   rB   rB   J   s    	2	B	2	B !*H *H6h&0""288RXX6	5	B	5	B99R##E**r   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   multiplyr#   s      r   rF   rF   Z   r@   r   c                     t        |      }t        |       } t        | j                        }d|v s|dk(  r!t	        j
                  | j                  d      }n|}t        j                  | ||      j                  |      S )Nr   boolr/   axiskeepdims)	r	   r   r   r   r   r   r   meanr2   )r*   rJ   rK   	ori_dtyper=   s        r   rL   rL   h   sl    %d+D!A!!''*I	Y&0))!''9= 7714(3::<HHr   c                 J    t        |      }t        j                  | |||      S N)rJ   rK   initial)r	   r   maxr*   rJ   rK   rP   s       r   rQ   rQ   s   !    %d+D66!$7CCr   c                 `    |xs t        j                         }t        j                  | |      S Nr   )r   floatxr   onesshaper   s     r   rX   rX   x   s#    $V]]_E775&&r   c                 `    |xs t        j                         }t        j                  | |      S rU   )r   rW   r   zerosrY   s     r   r\   r\   }   #    $V]]_E88E''r   c                 ,    t        j                  |       S r(   )r   absoluter*   s    r   r_   r_          ;;q>r   c                     t        |       S r(   )r_   r`   s    r   absrc      s    A;r   c                     t        |       } t        | j                        }d|v s|dk(  r#| j                  t	        j
                               } t        j                  |       S Nr   rH   )r   r   r   r2   r   rW   r   fabsr*   r   s     r   rf   rf      sI    !Aagg&E~&HHV]]_%771:r   c                 H    t        |      }t        j                  | ||      S NrI   )r	   r   allr*   rJ   rK   s      r   rj   rj          %d+D66!$22r   c                 6    t        j                  | ||||      S )N)rtolatol	equal_nan)r   allcloser$   r%   rn   ro   rp   s        r   rq   rq      s    ;;r2DtyIIr   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S Nint64)r   r   r   r   rW   r   r   r   r2   r   anglerg   s     r   rv   rv      sY    !A!W,""177E2	A88A;r   c                 H    t        |      }t        j                  | ||      S ri   )r	   r   anyrk   s      r   rx   rx      rl   r   c                 H    t        |      }t        j                  | ||      S ri   )r	   r   amaxrk   s      r   rz   rz          %d+D7714(33r   c                 H    t        |      }t        j                  | ||      S ri   )r	   r   aminrk   s      r   r}   r}      r{   r   c                    t        |      }t        |       } t        |      }t        j                  | j                  |j                        }| j                  |      } |j                  |      }t        j                  | ||      S NrJ   )r	   r   r   r   r   r2   r   append)r$   r%   rJ   r   s       r   r   r      si    %d+D	2	B	2	Brxx2E	5	B	5	B99R$''r   c           	      :   |wt        | dt        |             g}|%|j                  t        |dt        |                   |%|j                  t        |dt        |                   t        j                  | }|d| }} |d}t        j                  | |||      S )Nr   r   r   )stepr   )r    r!   r   r   r   r   arange)startstopr   r   r<   s        r   r   r      s    }$UGT%[AB$$WT7DJ%GH$$WT7DJ%GH""$56|t|99UDt599r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   arccosrg   s     r   r   r      Y    !A!W,""177E2	A99Q<r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   arccoshrg   s     r   r   r      Y    !A!W,""177E2	A::a=r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   arcsinrg   s     r   r   r      r   r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   arcsinhrg   s     r   r   r      r   r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   arctanrg   s     r   r   r      r   r   c                     t        |       } t        |      }t        j                  | j                  |j                  t              }| j                  |      } |j                  |      }t        j                  | |      S r(   )r   r   r   r   r   r2   r   arctan2r#   s      r   r   r      s_    	2	B	2	Brxx59E	5	B	5	B::b"r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   arctanhrg   s     r   r   r     r   r   c                    t        |       } t        |      }t        | j                        }d|vs| j                  dk(  r't        j                  | ||      j                  d      S t        j                  |d      }| j                  |      } | dk(  t        j                  |       z  }t        j                  |t        j                  | j                        j                   |       } t        j                  | ||      j                  d      S Nr   r   rI   r-   r/           )r   r	   r   r   r   r   argmaxr2   r   r   signbitwherefinfotinyr*   rJ   rK   r   is_negative_zeros        r   r   r         !A%d+Dagg&Eeqvv{yy9@@IIui0E	ASBJJqM1
!BHHQWW$5$:$:#:A>A99QTH5<<WEEr   c                    t        |       } t        |      }t        | j                        }d|vs| j                  dk(  r't        j                  | ||      j                  d      S t        j                  |d      }| j                  |      } | dk(  t        j                  |       z  }t        j                  |t        j                  | j                        j                   |       } t        j                  | ||      j                  d      S r   )r   r	   r   r   r   r   argminr2   r   r   r   r   r   r   r   s        r   r   r      r   r   c                 d    t        |      }t        j                  | |      j                  d      S Nr   r-   )r	   r   argsortr2   r*   rJ   s     r   r   r   .  s(    %d+D::ad#**733r   c                     t        | |      S rU   r
   rg   s     r   r   r   3  s    Qe,,r   c                 <    t        |       } | j                  |      S rU   )r   viewrg   s     r   r   r   7  s    !A666r   c                 >   t        |      }t        |       } | j                  t        g}|&t        |      }|j	                  |j                         t        j                  | }| j                  |      } ||j                  |      }t        j                  | ||      S )N)weightsrJ   )
r	   r   r   r   r   r   r   r2   r   average)r*   rJ   r   r<   r   s        r   r   r   <  s    %d+D!A%(#G,  / 12E	A..'::at44r   c                     t        |       } t        j                  |       j                  t	        j
                               S r(   )r   r   bartlettr2   r   rW   r`   s    r   r   r   J  +    !A;;q>  11r   c                     t        |       } t        j                  |       j                  t	        j
                               S r(   )r   r   hammingr2   r   rW   r`   s    r   r   r   O  +    !A::a=00r   c                     t        |       } t        j                  |       j                  t	        j
                               S r(   )r   r   hanningr2   r   rW   r`   s    r   r   r   T  r   r   c                 
   t        |       } t        |      }t        j                  | j                  |j                        }|dv rt	        j
                         }n|dv rd}t        j                  | |      j                  |      S N)r,   int16r-   uint8uint16uint32)ru   float64)	r   r   r   r   r   rW   r   	heavisider2   r#   s      r   r   r   Y  sm    	2	B	2	Brxx2EGG	)	<<B&&u--r   c                     t        |       } t        j                  | |      j                  t	        j
                               S r(   )r   r   kaiserr2   r   rW   )r*   betas     r   r   r   f  s.    !A99Q$$V]]_55r   c           	         |rt        d      t        |       } | j                  g}|9t        |      }|j                  |j                         t	        j
                  | }nd}t        | j                        dk(  re|fd}t        t        ||             }n$fd}t        t        |t        | |                  }t        j                  |      j                  |      S t        j                  | |      j                  |      S )Nz2Unsupported value `sparse=True` with numpy backendr-   r   c                 2    t        j                  |       S )N)	minlengthr   bincount)arrr   s    r   bincount_fnzbincount.<locals>.bincount_fny  s    {{3)<<r   c                 @    t        j                  | d   | d         S )Nr   r   )r   r   r   )arr_wr   s    r   r   zbincount.<locals>.bincount_fn  s#    {{!HeAh) r   )r   r   r   r   r   r   r   rZ   r6   mapzipr   stackr2   r   )r*   r   r   sparser<   r   r   	bincountss     `     r   r   r   k  s    MNN!A	#G,  /""$56
177|q?= Sa01I
 Sc!Wo>?Ixx	"))%00;;q'9-44U;;r   c                     t        |       } t        |      }t        j                  | j                  |j                        }| j	                  |      } |j	                  |      }t        j                  | |      S r(   )r   r   r   r   r2   r   bitwise_andr*   yr   s      r   r   r     [    !A!Aqww0E	A	A>>!Qr   c                 B    t        |       } t        j                  |       S r(   )r   r   bitwise_notr`   s    r   bitwise_invertr     s    !A>>!r   c                     t        |       S r(   )r   r`   s    r   r   r     s    !r   c                     t        |       } t        |      }t        j                  | j                  |j                        }| j	                  |      } |j	                  |      }t        j                  | |      S r(   )r   r   r   r   r2   r   
bitwise_orr   s      r   r   r     s[    !A!Aqww0E	A	A==Ar   c                     t        |       } t        |      }t        j                  | j                  |j                        }| j	                  |      } |j	                  |      }t        j                  | |      S r(   )r   r   r   r   r2   r   bitwise_xorr   s      r   r   r     r   r   c                    t        |       } t        |t              sWt        |      }t        j                  | j
                  |j
                        }| j                  |      } |j                  |      }t        j                  | |      S r(   )	r   r   r   r   r   r   r2   r   
left_shiftr   s      r   bitwise_left_shiftr     se    !Aaa ""177AGG4HHUOHHUO==Ar   c                     t        | |      S r(   )r   r*   r   s     r   r   r     s    a##r   c                    t        |       } t        |t              sWt        |      }t        j                  | j
                  |j
                        }| j                  |      } |j                  |      }t        j                  | |      S r(   )	r   r   r   r   r   r   r2   r   right_shiftr   s      r   bitwise_right_shiftr     se    !Aaa ""177AGG4HHUOHHUO>>!Qr   c                     t        | |      S r(   )r   r   s     r   r   r     s    q!$$r   c                     t        |       } t        j                  |       j                  t	        j
                               S r(   )r   r   blackmanr2   r   rW   r`   s    r   r   r     r   r   c                 .    t        j                  | |      S r(   )r   broadcast_to)r*   rZ   s     r   r   r     s    ??1e$$r   c                     t        |       } t        | j                        }|dv rt        j                         }n|dk(  rd}t        j                  |       j                  |      S )N)rH   r,   r   r-   r   r   r   ru   r   )r   r   r   r   rW   r   cbrtr2   rg   s     r   r   r     sT    !Aagg&EOO	'	771:U##r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   ceilrg   s     r   r   r     Y    !A!W,""177E2	A771:r   c                     t        |       } t        | j                        }|dk(  rd}t        j                  | ||      j                  |      S NrH   r-   )r   r   r   r   clipr2   )r*   x_minx_maxr   s       r   r   r     sD    !Aagg&E771eU#**511r   c                    t        |      }t        | D cg c]  }t        |dt        |             c}      }t	        |      dkD  r+t        j                  | t        j                  fd|       } t        j                  | |      S c c}w )Nr   r   c                 8    t        |       j                        S r(   r   r2   rg   s    r   r4   zconcatenate.<locals>.<lambda>      '*11%8 r   r   )r	   r7   r    r!   r   r   r   r   r5   r   concatenate)xsrJ   r*   	dtype_setr   s       @r   r   r     sx    %d+D2>aWQa1>?I
9~""I.8"
 >>"4(( ?   B
c                 ,    t        j                  |       S r(   )r   	conjugater`   s    r   r   r     s    <<?r   c                     t        |       S r(   )r   r`   s    r   conjr      s    Q<r   c                 ,    t        j                  |       S r(   )r   copyr`   s    r   r   r         771:r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   cosrg   s     r   r  r    Y    !A!W,""177E2	A66!9r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   coshrg   s     r   r  r    r   r   c                 v    t        |      }t        t        j                  | |            j	                  d      S r   )r	   r   r   count_nonzeror2   r   s     r   r  r    s1    %d+D R--ad;<CCGLLr   c                    t        |      }t        |       } t        |      }t        j                  | j                  |j                        }| j                  |      } |j                  |      }t        j                  | |||||      S )N)axisaaxisbaxiscrJ   )r	   r   r   r   r   r2   r   cross)r$   r%   r
  r  r  rJ   r   s          r   r  r  #  sy    %d+D	2	B	2	Brxx2E	5	B	5	B88

 r   c                     t        |      }t        j                  |xs | j                        }|dk(  rd}t	        j
                  | ||      S NrH   r-   )rJ   r   )r	   r   r   r   r   cumprodr*   rJ   r   s      r   r  r  4  sD    %d+Du/0E::ad%00r   c                     t        |      }t        j                  |xs | j                        }|dk(  rd}t	        j
                  | ||      S r  )r	   r   r   r   r   cumsumr  s      r   r  r  <  sD    %d+Du/0E99QT//r   c                     t        |       } | j                  dv rd}n/| j                  dv r| j                  }nt        j                         }t	        j
                  |       j                  |      S Nru   r   r   )r.   float16)r   r   r   rW   r   deg2radr2   rg   s     r   r  r  D  X    !Aww&&	
+	+::a=&&r   c                     t        |       } | j                  dv rd}n/| j                  dv r| j                  }nt        j                         }t	        j
                  |       j                  |      S r  )r   r   r   rW   r   rad2degr2   rg   s     r   r  r  Q  r  r   c                 0    t        j                  | |      S N)r   )r   diagr*   r   s     r   r  r  ^      771?r   c                 0    t        j                  | |      S r  )r   diagflatr  s     r   r"  r"  b  s    ;;qAr   c                 `    t        |      }t        |      }t        j                  | |||      S )N)offsetaxis1axis2)r	   r   diagonal)r*   r$  r%  r&  s       r   r'  r'  f  s+    &u-E&u-E;;quEBBr   c                 2    t        j                  | ||      S )N)nrJ   )r   diff)ar)  rJ   s      r   r*  r*  l  s    771%%r   c                 h    t        j                  | |      j                  t         j                        S r(   )r   digitizer2   r-   )r*   binss     r   r-  r-  p  s"    ;;q$&&rxx00r   c                     t        |       } t        |      }t        j                  | j                  |j                        }| j	                  |      } |j	                  |      }t        j                  | |      S r(   )r   r   r   r   r2   r   dotr#   s      r   r0  r0  t  s\    	2	B	2	Brxx2E	5	B	5	B66"b>r   c                    t        | D cg c]  }t        |dt        |             c}      }t        |      dkD  r+t	        j
                  | t        j                  fd|       } t        j                  |       S c c}w )Nr   r   c                 8    t        |       j                        S r(   r   rg   s    r   r4   zdstack.<locals>.<lambda>  r   r   )
r7   r    r!   r   r   r   r   r5   r   dstackr   r*   r   r   s      @r   r3  r3  }  k    2>aWQa1>?I
9~""I.8"
 99R= ?   A=c                 `    |xs t        j                         }t        j                  | |      S rU   )r   rW   r   emptyrY   s     r   r8  r8    r]   r   c                 0    t        j                  | |      S rU   )r   
empty_likerg   s     r   r:  r:        ==%((r   c                 .    t        j                  | |      S r(   )r   equalr$   r%   s     r   r=  r=    s    88Br   c                     t        |       } t        | j                        }d|v s|dk(  r#| j                  t	        j
                               } t        j                  |       S re   )r   r   r   r2   r   rW   r   expr*   rM   s     r   r@  r@    sK    !A!!''*I	Y&0HHV]]_%66!9r   c                     t        |       } t        | j                        }d|v s|dk(  r#| j                  t	        j
                               } t        j                  |       S re   )r   r   r   r2   r   rW   r   exp2rA  s     r   rC  rC    sK    !A!!''*I	Y&0HHV]]_%771:r   c                 D    t        |      }t        j                  | |      S r(   )r	   r   expand_dimsr   s     r   rE  rE    s    %d+D>>!T""r   c                     t        |       } t        | j                        }d|v s|dk(  r#| j                  t	        j
                               } t        j                  |       S re   )r   r   r   r2   r   rW   r   expm1rA  s     r   rG  rG    sK    !A!!''*I	Y&0HHV]]_%88A;r   c                 F    t        |      }t        j                  | |      S r   )r	   r   flipr   s     r   rI  rI        %d+D7714  r   c                 ,    t        j                  |       S r(   )r   fliplrr`   s    r   rL  rL        99Q<r   c                 ,    t        j                  |       S r(   )r   flipudr`   s    r   rO  rO    rM  r   c                    t        |       } t        | j                        dk(  rt        j                         n#t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   floorrg   s     r   rQ  rQ    s`    !A QWW%0 	/ 

 	
A88A;r   c                 b    |xs t        j                         }t        j                  | ||      S rU   )r   rW   r   full)rZ   
fill_valuer   s      r   rS  rS    s%    $V]]_E775*E22r   c                 2    t        j                  | ||      S rU   )r   	full_like)r*   rT  r   s      r   rV  rV    s    <<:U33r   c                     t        |       } t        |      }t        j                  | j                  |j                        }t	        j
                  | |      j                  |      S r(   )r   r   r   r   r   gcdr2   r#   s      r   rX  rX    sK    	2	B	2	Brxx2E66"b>  ''r   c                 h    |xs t        j                         }t        j                  | |||||      S )N)numendpointr   rJ   )r   rW   r   	geomspace)r   r   rZ  r[  r   rJ   s         r   r\  r\    s0    $V]]_E<<txu4 r   c                 .    t        j                  | |      S r(   )r   greaterr>  s     r   r^  r^    s    ::b"r   c                 .    t        j                  | |      S r(   )r   greater_equalr>  s     r   r`  r`    s    B##r   c                    t        | D cg c]  }t        |dt        |             c}      }t        |      dkD  r+t	        j
                  | t        j                  fd|       } t        j                  |       S c c}w )Nr   r   c                 8    t        |       j                        S r(   r   rg   s    r   r4   zhstack.<locals>.<lambda>  r   r   )
r7   r    r!   r   r   r   r   r5   r   hstackr4  s      @r   rc  rc    r5  r6  c                 D    t        |       } t        j                  | |      S r(   )r   r   hsplitr*   indices_or_sectionss     r   re  re        !A99Q+,,r   c                 
   t        |       } t        |      }t        j                  | j                  |j                        }|dv rt	        j
                         }n|dv rd}t        j                  | |      j                  |      S r   )	r   r   r   r   r   rW   r   hypotr2   r#   s      r   rj  rj    sm    	2	B	2	Brxx2EGG	)	88B""5))r   c                 `    |xs t        j                         }t        j                  | |      S rU   )r   rW   r   identity)r)  r   s     r   rl  rl    s#    $V]]_E;;q&&r   c                 ,    t        j                  |       S r(   )r   imagr`   s    r   rn  rn  
  r  r   c                     t        |       } t        | j                        dv rdn#t        j                  | j                  t
              }| j                  |      } t        j                  |       S )Nr  r   )	r   r   r   r   r   r   r2   r   i0rg   s     r   rp  rp    sZ    !A QWW%)== 	/ 

 	
A558Or   c                 4    t        j                  | ||||      S r(   )r   iscloserr   s        r   rr  rr    s    ::b"dD)44r   c                 ,    t        j                  |       S r(   )r   isfiniter`   s    r   rt  rt    ra   r   c                 `    t        |       } t        |      }t        j                  | |||      S )N)assume_uniqueinvert)r   r   isin)r$   r%   rv  rw  s       r   rx  rx  !  s+    	2	B	2	B772rvFFr   c                 ,    t        j                  |       S r(   )r   isinfr`   s    r   rz  rz  '      88A;r   c                 ,    t        j                  |       S r(   )r   isnanr`   s    r   r}  r}  +  r{  r   c                 B    t        |       } t        j                  |       S r(   )r   r   isneginfr`   s    r   r  r  /      !A;;q>r   c                 B    t        |       } t        j                  |       S r(   )r   r   isposinfr`   s    r   r  r  4  r  r   c                 B    t        |       } t        j                  |       S r(   )r   r   isrealr`   s    r   r  r  9  s    !A99Q<r   c                     t        |       } t        |      }t        j                  | j                  |j                        }t	        j
                  | |      j                  |      S r(   )r   r   r   r   r   kronr2   r#   s      r   r  r  >  sK    	2	B	2	Brxx2E772r?!!%((r   c                     t        |       } t        |      }t        j                  | j                  |j                        }t	        j
                  | |      j                  |      S r(   )r   r   r   r   r   lcmr2   r#   s      r   r  r  E  sK    	2	B	2	Brxx2E66"b>  ''r   c                 P   t        |       } t        |      }t        j                  | j                  |j                  t              }t        |j                        t        j                  vrt        d|j                         t        j                  | |      j                  |      S )Nz;ldexp exponent must be an integer type. Received: x2 dtype=)r   r   r   r   r   r   	INT_TYPES	TypeErrorr   ldexpr2   r#   s      r   r  r  L  s    	2	B	2	Brxx59E"&*:*::""$((-
 	
 88B""5))r   c                 .    t        j                  | |      S r(   )r   lessr>  s     r   r  r  Y  s    772r?r   c                 .    t        j                  | |      S r(   )r   
less_equalr>  s     r   r  r  ]      ==R  r   c           	          t        |      }|Ct        | dt        |             t        |dt        |            t        g}t	        j
                  | }t        j                  | ||||||      S )Nr   )rZ  r[  retstepr   rJ   )r	   r    r!   r   r   r   r   linspace)r   r   rZ  r[  r  r   rJ   r<   s           r   r  r  a  sw     &d+D}E7DK0D'4:.

 ""$56;; r   c                     t        |       } t        | j                        dk(  rt        j                         n#t        j                  | j                  t              }t        j                  | |      S Nru   rV   )
r   r   r   r   rW   r   r   r   r   logrg   s     r   r  r  w  sV    !A QWW%0 	/ 

 66!5!!r   c                     t        |       } t        | j                        dk(  rt        j                         n#t        j                  | j                  t              }t        j                  | |      S r  )
r   r   r   r   rW   r   r   r   r   log10rg   s     r   r  r    V    !A QWW%0 	/ 

 88AU##r   c                     t        |       } t        | j                        dk(  rt        j                         n#t        j                  | j                  t              }t        j                  | |      S r  )
r   r   r   r   rW   r   r   r   r   log1prg   s     r   r  r    r  r   c                     t        |       } t        | j                        dk(  rt        j                         n#t        j                  | j                  t              }t        j                  | |      S r  )
r   r   r   r   rW   r   r   r   r   log2rg   s     r   r  r    sV    !A QWW%0 	/ 

 771E""r   c                     t        |       } t        |      }t        j                  | j                  |j                  t              }| j                  |      } |j                  |      }t        j                  | |      S r(   )r   r   r   r   r   r2   r   	logaddexpr#   s      r   r  r    s_    	2	B	2	Brxx59E	5	B	5	B<<Br   c                     t        |       } t        |      }t        j                  | j                  |j                  t              }t        j                  | |      j                  |      S r(   )r   r   r   r   r   r   
logaddexp2r2   r#   s      r   r  r    sN    	2	B	2	Brxx59E==R ''..r   c                 .    t        j                  | |      S r(   )r   logical_andr>  s     r   r  r        >>"b!!r   c                 ,    t        j                  |       S r(   )r   logical_notr`   s    r   r  r    s    >>!r   c                 .    t        j                  | |      S r(   )r   
logical_orr>  s     r   r  r    r  r   c           	          |Ct        | dt        |             t        |dt        |            t        g}t        j                  | }t        j                  | ||||||      S )Nr   )rZ  r[  baser   rJ   )r    r!   r   r   r   r   logspace)r   r   rZ  r[  r  r   rJ   r<   s           r   r  r    sk    }E7DK0D'4:.

 ""$56;; r   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   maximumr#   s      r   r  r        b3,'r"b3,'r"GT"X&GT"X&E 
2u	%B	2u	%B::b"r   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   fmaxr#   s      r   r  r        b3,'r"b3,'r"GT"X&GT"X&E 
2u	%B	2u	%B772r?r   c                     t        j                  | j                  t              }t	        j
                  | ||      j                  |      S ri   )r   r   r   r   r   medianr2   r*   rJ   rK   r   s       r   r  r    s6    qww.E99QTH5<<UCCr   xy)indexingc                 ,    t        j                  |d| iS )Nr  )r   meshgrid)r  r*   s     r   r  r    s    ;;-H--r   c                 J    t        |      }t        j                  | |||      S rO   )r	   r   minrR   s       r   r  r    rS   r   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   minimumr#   s      r   r  r    r  r   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   fminr#   s      r   r  r    r  r   c                     t        |       } t        |      }t        j                  | j                  |j                        }|dk(  rd}| j	                  |      } |j	                  |      }t        j                  | |      S r   )r   r   r   r   r2   r   modr#   s      r   r  r    sh    	2	B	2	Brxx2E	5	B	5	B66"b>r   c                     t        |       } t        |      }t        j                  | j                  |j                        }|dk(  rd}| j	                  |      } |j	                  |      }t        j                  | |      S r   )r   r   r   r   r2   r   fmodr#   s      r   r  r    sh    	2	B	2	Brxx2E	5	B	5	B772r?r   c                 2    t        j                  | ||      S )N)sourcedestination)r   moveaxis)r*   r  r  s      r   r  r  *  s    ;;q[AAr   c                    t        j                  | j                  t         j                        st	        | ||      S t        j
                  |       }t        j                  t        j                  |||      dt        j                  t        j                  |t         j                   |       ||      j                  d            S NrI   r-   )r   
issubdtyper   floatingr   r}  r   rj   	nanargmaxinfr2   r*   rJ   rK   nan_masks       r   r  r  .  s    =="++.adX66xx{H88
xdX6

HHXw*	

&/ r   c                    t        j                  | j                  t         j                        st	        | ||      S t        j
                  |       }t        j                  t        j                  |||      dt        j                  t        j                  |t         j                  |       ||      j                  d            S r  )r   r  r   r  r   r}  r   rj   	nanargminr  r2   r  s       r   r  r  =  s    =="++.adX66xx{H88
xdX6

HHXrvvq)x	

&/ r   c                     t        |      }t        j                  |xs | j                        }|dk(  rd}t	        j
                  | ||      S r  )r	   r   r   r   r   	nancumsumr  s      r   r  r  L  sD    %d+Du/0E<<E22r   c                     t        |      }t        j                  |xs | j                        }|dk(  rd}t	        j
                  | ||      S r  )r	   r   r   r   r   
nancumprodr  s      r   r  r  T  sD    %d+Du/0E==U33r   c                 2    t        j                  | ||      S ri   )r   nanmaxrk   s      r   r  r  \      99QTH55r   c                     t        j                  t        | j                        t              }t        j                  | ||      j                  |      S ri   )r   r   r   r   r   r   nanmeanr2   r  s       r   r  r  `  s<    095AE::adX6==eDDr   c                     t        j                  t        | j                        t              }t        j                  | ||      j                  |      S ri   )r   r   r   r   r   r   	nanmedianr2   r  s       r   r  r  e  s<    095AE<<x8??FFr   c                 2    t        j                  | ||      S ri   )r   nanminrk   s      r   r  r  j  r  r   c                    t        |       } t        |      }t        | j                        }|dk(  r#| j                  t	        j
                               } t        | j                        dk(  r#| j                  t	        j
                               } t        j                  | j                        st	        j
                         }n| j                  }t        j                  | ||||      j                  |      S NrH   rJ   methodrK   )
r   r   r   r2   r   rW   r   is_float_dtyper   nanpercentiler*   qrJ   r  rK   rM   r   s          r   r  r  n  s    !A!A!!''*IFHHV]]_%!V+HHV]]_%!!!''*	14fUmr   c                     t        |      }t        |       } t        j                  | j                        }|dv rd}n|dv rd}t        j                  | |||      S NrH   r,   r   r-   r   r   r   rJ   rK   r   )r	   r   r   r   r   r   nanprodr  s       r   r  r    s[    %d+D!Aqww'E))	%	%::adXUCCr   c                 ,   t        |       } t        | j                        }|dk(  r#| j                  t	        j
                               } t        j                  | j                  t              }t        j                  | ||||      j                  |      S r  )r   r   r   r2   r   rW   r   r   r   r   nanquantiler  s          r   r  r    sq    !A!!''*IFHHV]]_%qww.E>>	14fUmr   c                    t        |      }t        |       } t        j                  | j                  d      }t        j                  | j                  t
              }t        j                  | |||      j                  |      S Nr/   r  )	r	   r   r   r   r   r   r   nanstdr2   r*   rJ   rK   r3   r=   s        r   r  r    f    %d+D!A&&qww	:M%%aggu5L99	x}f\r   c                     t        |      }t        | j                        }|dv rd}n|dv rd}t        j                  | ||      j                  |      S Nr  r-   r  r   rI   )r	   r   r   r   nansumr2   r  s       r   r  r    sT    %d+Dagg&E))	%	%99QTH5<<UCCr   c                    t        |      }t        |       } t        j                  | j                  d      }t        j                  | j                  t
              }t        j                  | |||      j                  |      S r  )	r	   r   r   r   r   r   r   nanvarr2   r  s        r   r  r    r  r   c                 4    t        j                  | |||      S )N)nanposinfneginf)r   
nan_to_num)r*   r  r  r  s       r   r  r    s    ==F6BBr   c                 ,    t        j                  |       S r(   )r   r   r`   s    r   r   r     r  r   c                 L    t        d t        j                  |       D              S )Nc              3   >   K   | ]  }|j                  d         yw)r-   Nr1   )r)   indicess     r   r+   znonzero.<locals>.<genexpr>  s     FW(Fs   )tupler   nonzeror`   s    r   r  r    s    F

1FFFr   c                 .    t        j                  | |      S r(   )r   	not_equalr>  s     r   r  r    s    <<Br   c                 0    t        j                  | |      S rU   )r   
zeros_likerg   s     r   r  r    r;  r   c                 0    t        j                  | |      S rU   )r   	ones_likerg   s     r   r
  r
    s    <<''r   c                     t        |       } t        |      }t        j                  | j                  |j                        }| j	                  |      } |j	                  |      }t        j                  | |      S r(   )r   r   r   r   r2   r   outerr#   s      r   r  r    ]    	2	B	2	Brxx2E	5	B	5	B88Br   c                 l    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   pad)r*   	pad_widthr  r  r;   s        r   r  r    sY    F":""&) 
 %4 !66!Y4T4V44r   c                 B   t        |       } t        |      }t        | j                        }|dk(  r#| j                  t	        j
                               } t        j                  | j                  t              }t        j                  | ||||      j                  |      S r  )r   r   r   r2   r   rW   r   r   r   r   
percentiler  s          r   r  r    s{    !A!A!!''*IFHHV]]_%qww.E==	14fUmr   c                     t        |      }t        |       } |,t        j                  | j                        }|dv rd}n|dv rd}t        j                  | |||      S r  )r	   r   r   r   r   r   prodr  s       r   r  r    s`    %d+D!A}""177+--E))E7714(%@@r   c                 2    t        j                  | ||      S ri   )r   ptprk   s      r   r  r    s    66!$22r   c                 v   t        |      }t        |       } t        | j                        }|dk(  r#| j	                  t        j                               } |dk(  rt        j                         }n$t        j                  | j                  t              }t        j                  | ||||      j	                  |      S )NrH   ru   r  )r	   r   r   r   r2   r   rW   r   r   r   r   quantiler  s          r   r  r     s    %d+D!A!!''*IFHHV]]_%G""177E2;;	14fUmr   c                 ,    t        j                  |       S r(   )r   ravelr`   s    r   r  r    r{  r   c                     t        j                  | j                        t        fdt	        j
                  | |      D              S )Nc              3   @   K   | ]  }|j                          y wr(   r1   )r)   r  r   s     r   r+   z unravel_index.<locals>.<genexpr>  s      ")us   )r   r   r   r  r   unravel_index)r  rZ   r   s     @r   r   r     s>    w}}-E -/-=-=gu-M  r   c                 ,    t        j                  |       S r(   )r   realr`   s    r   r"  r"    r  r   c                 ,    t        j                  |       S r(   )r   
reciprocalr`   s    r   r$  r$     s    ==r   c                 2    t        j                  | ||      S r   )r   repeat)r*   repeatsrJ   s      r   r&  r&  $  s    99Qd++r   c                 .    t        j                  | |      S r(   )r   reshape)r*   newshapes     r   r)  r)  (  s    ::a""r   c                 2    t        j                  | ||      S r   )r   roll)r*   shiftrJ   s      r   r,  r,  ,  s    771e$''r   c                    t        |       dk7  rt        d| j                         | j                  d   t        j                  t        j
                        j                  k  rdnd}t        j                  | ||      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=r   r-   ru   )side)	r   r   rZ   r   iinfor-   rQ   searchsortedr2   )sorted_sequencevaluesr/  out_types       r   r1  r1  0  s    O!% &5%:%:$;=
 	
   #rxx'9'='== 	 
 ???F>EEhOOr   c                 ,    t        j                  |       S r(   )r   signr`   s    r   r6  r6  @  r  r   c                 ,    t        j                  |       S r(   )r   r   r`   s    r   r   r   D  s    ::a=r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   sinrg   s     r   r9  r9  H  r  r   c                 $   t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       j                  |      S rt   )r   r   r   r   rW   r   r   r   r2   r   sincrg   s     r   r;  r;  R  sd    !A!W,""177E2	A771:U##r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   sinhrg   s     r   r=  r=  \  r   r   c                 ,    t        j                  |       S r(   )r   sizer`   s    r   r?  r?  f  r  r   c                 F    t        |      }t        j                  | |      S r   )r	   r   sortr   s     r   rA  rA  j  rJ  r   c                 H    t        |      }t        j                  | ||      S r   )r	   r   splitr*   rg  rJ   s      r   rC  rC  o  s     %d+D88A*66r   c                 H    t        |      }t        j                  | ||      S r   )r	   r   array_splitrD  s      r   rF  rF  t  s     %d+D>>!0t<<r   c                    t        |      }t        | D cg c]  }t        |dt        |             c}      }t	        |      dkD  r+t        j                  | t        j                  fd|       } t        j                  | |      S c c}w )Nr   r   c                 8    t        |       j                        S r(   r   )r+  r   s    r   r4   zstack.<locals>.<lambda>~  s    ):1)=)D)DU)K r   r   )r	   r7   r    r!   r   r   r   r   r5   r   r   )r*   rJ   r+  r   r   s       @r   r   r   y  st    %d+D1=aWQa1=>I
9~""I.KQO88AD!!	 >r   c                     t        |      }t        |       } t        | j                        }d|v s|dk(  r#| j	                  t        j                               } t        j                  | ||      S )Nr   rH   rI   )	r	   r   r   r   r2   r   rW   r   std)r*   rJ   rK   rM   s       r   rJ  rJ    sZ    %d+D!A!!''*I	Y&0HHV]]_%66!$22r   c                 2    t        j                  | ||      S )N)r%  r&  )r   swapaxes)r*   r%  r&  s      r   rL  rL    s    ;;qU33r   c                 H    t        |      }t        j                  | ||      S r   )r	   r   taker*   r  rJ   s      r   rN  rN    s    %d+D771gD))r   c                 H    t        |      }t        j                  | ||      S r   )r	   r   take_along_axisrO  s      r   rQ  rQ    s!    %d+Dat44r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   tanrg   s     r   rS  rS    r  r   c                    t        |       } t        | j                        dk(  rt        j                         }n$t        j                  | j                  t              }| j                  |      } t        j                  |       S rt   )r   r   r   r   rW   r   r   r   r2   r   tanhrg   s     r   rU  rU    r   r   c                 0   t        |t              rt        |      n|}t        |       } t        |      }t	        j
                  | j                  |j                        }| j                  |      } |j                  |      }t        j                  | ||      S N)r   )
r   r6   r  r   r   r   r   r2   r   	tensordot)r$   r%   r   r   s       r   rX  rX    ss    $T405;dD	2	B	2	Brxx2E	5	B	5	B<<BT**r   c                 0    t        j                  | |      S )N)decimals)r   round)r*   rZ  s     r   r[  r[    s    88A))r   c                 .    t        j                  | |      S r(   )r   tile)r*   r'  s     r   r]  r]    s    771gr   c                     t        |      }t        |      }t        |       } t        | j                        }|dv rd}n|dv rd}t	        j
                  | ||||      S )Nr  r-   r  r   )r$  r%  r&  r   )r	   r   r   r   r   trace)r*   r$  r%  r&  r   s        r   r_  r_    sa    &u-E&u-E!Aagg&E))	%	%88AfEeLLr   c                 d    |xs t        j                         }t        j                  | |||      S N)Mr   r   )r   rW   r   triNrb  r   r   s       r   rc  rc    '    $V]]_E66!qAU++r   c                 0    t        j                  | |      S r  )r   trilr  s     r   rh  rh    r   r   c                 0    t        j                  | |      S r  )r   triur  s     r   rj  rj    r   r   c                     t        |       } t        | j                        }d|v sd|k(  r| S t        j                  |       S re   )r   r   r   r   truncrg   s     r   rl  rl    s:    !Aagg&E~588A;r   c                     t        |       } t        |      }t        j                  | j                  |j                        }| j	                  |      } |j	                  |      }t        j                  | |      S r(   )r   r   r   r   r2   r   vdotr#   s      r   rn  rn    s\    	2	B	2	Brxx2E	5	B	5	B772r?r   c                     t        |       } t        |      }t        j                  | j                  |j                        }| j	                  |      } |j	                  |      }t        j                  | |      S r(   )r   r   r   r   r2   r   innerr#   s      r   rp  rp    r  r   c                    t        | D cg c]  }t        |dt        |             c}      }t        |      dkD  r+t	        j
                  | t        j                  fd|       } t        j                  |       S c c}w )Nr   r   c                 8    t        |       j                        S r(   r   rg   s    r   r4   zvstack.<locals>.<lambda>  r   r   )
r7   r    r!   r   r   r   r   r5   r   vstackr4  s      @r   rs  rs    r5  r6  c                 D    t        |       } t        j                  | |      S r(   )r   r   vsplitrf  s     r   ru  ru    rh  r   excluded	signaturec                2    t        j                  | ||      S )Nrv  )r   	vectorize)pyfuncrw  rx  s      r   rz  rz    s    <<YGGr   c           
         ||t        |t        t        f      st        |      }t        |t        t        f      st        |      }t	        j
                  t        |dt        |            t        |dt        |                  }t        ||      }t        ||      }t        j                  | ||      S t        j                  |       S r   )
r   r   r   r   r   r   r    r!   r   r   )	conditionr$   r%   r   s       r   r   r     s    	~"."sEl+"2&B"sEl+"2&B""Bb*Bb*
 r5)r5)xx	2r**xx	""r   c           
      h   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |            t              }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   divider#   s      r   r  r    s    b3,'r"b3,'r"GT"X&GT"X&E
 
2u	%B	2u	%B99Rr   c           
         t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |            t              }t        | |      } t        ||      }t        j                  |dk(  t        j                  d|      t        j                  | |            S )Nr   r   rV   )r   r   r   r   r   r   r    r!   r   r   r   r  r#   s      r   divide_no_nanr  %  s    b3,'r"b3,'r"GT"X&GT"X&E
 
2u	%B	2u	%B 88B!GRXXau5ryyR7HIIr   c                     t        | |      S r(   )r  r>  s     r   true_divider  6  s    "b>r   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   powerr#   s      r   r  r  :  s    b3,'r"b3,'r"GT"X&GT"X&E 
2u	%B	2u	%B88Br   c                 ,    t        j                  |       S r(   )r   negativer`   s    r   r  r  H  ra   r   c                     t        |       } t        |      }t        j                  | j                  |j                  t              }t        j                  | |      j                  |      S r(   )r   r   r   r   r   r   	nextafterr2   r#   s      r   r  r  L  sN    	2	B	2	Brxx59E<<B&&u--r   c                     t        |       } t        | j                        dk(  r| j                  d      } t	        j
                  |       S r   )r   r   r   r2   r   squarer`   s    r   r  r  T  s9    !A!V+HHW99Q<r   c                     t        |       } t        | j                        dk(  rt        j                         n#t        j                  | j                  t              }t        j                  | |      S r  )
r   r   r   r   rW   r   r   r   r   sqrtrg   s     r   r  r  [  sV    !A QWW%0 	/ 

 771E""r   c                 F    t        |      }t        j                  | |      S r   )r	   r   squeezer   s     r   r  r  f  s    %d+D::ad##r   c                 j    t        |t              rt        |      n|}t        j                  | |      S rW  )r   r6   r  r   	transpose)r*   r   s     r   r  r  k  s'    $T405;dD<<%%r   c                     t        |       } t        j                  | j                  t              }|t        |      }t        |      }t        j                  | |||      j                  |      S )N)dxrJ   )r   r   r   r   r   r   	trapezoidr2   )r   r*   r  rJ   r=   s        r   r  r  p  s[    !A%%aggu5L}a 	2	B<<1$/66|DDr   c                 *   t        |       } t        j                  | j                        }t        j                  | j                  t	        j
                               }| j                  |      } t        j                  | ||      j                  |      S )N)re  
increasing)	r   r   r   r   r   rW   r2   r   vander)r*   re  r  r=   r3   s        r   r  r  y  sh    !A%%agg.L&&qww@M	A99Q!
3::<HHr   c                    t        |      }t        |       } t        j                  | j                  d      }t        j                  | j                  t
              }t        j                  | |||      j                  |      S r  )	r	   r   r   r   r   r   r   varr2   r  s        r   r  r    sg    %d+D!A&&qww	:M%%aggu5L66!$GNN r   c                     t        |      }t        | j                        }|dv rd}n|dv rd}t        j                  | ||      j                  |      S r  )r	   r   r   r   sumr2   r  s       r   r  r    sT    %d+Dagg&E))	%	%66!$299%@@r   c                 d    |xs t        j                         }t        j                  | |||      S ra  )r   rW   r   eyerd  s       r   r  r    rf  r   c           
      ^   t        | t        t        f      st        |       } t        |t        t        f      st        |      }t	        j
                  t        | dt        |             t        |dt        |                  }t        | |      } t        ||      }t        j                  | |      S r   )
r   r   r   r   r   r   r    r!   r   floor_divider#   s      r   r  r    s    b3,'r"b3,'r"GT"X&GT"X(FE 
2u	%B	2u	%B??2r""r   c                 .    t        j                  | |      S r(   )r   logical_xorr>  s     r   r  r    r  r   c                     | j                   dv rd}n/| j                   dv r| j                   }nt        j                         }t        |       } t	        j
                  |       j                  |      S r  )r   r   rW   r   r   corrcoefr2   rg   s     r   r  r    sX    ww&&	
+	+!A;;q>  ''r   c           
          t        j                  t        | dt        |             t        |dt        |                  }|dk(  rd}n|dvrd}t	        | |      } t	        ||      }t        j                  | ||      S )Nr   ru   r   )r.   r  r   r/   )r   r   r    r!   r   r   	correlate)r$   r%   r  r   s       r   r  r    sz    GT"X&GT"X&E 	8	8	2u	%B	2u	%B<<B%%r   c                 2    t        j                  | ||      S )N)default)r   select)condlist
choicelistr  s      r   r  r    s    99Xz7;;r   c                 R    t        t        j                  j                  |             S r(   )r  r   linalgslogdetr`   s    r   r  r    s    ""1%&&r   c                 N    t        j                  | ||      j                  d      S )Nr-   )r   argpartitionr2   )r*   kthrJ   s      r   r  r    s     ??1c4(//88r   c                 2    t        j                  | ||      S )N)r.  range)r   	histogram)r*   r.  r  s      r   r  r    s    <<E22r   c                    t        j                  | ||||d      }|s|s|s|g}nt        |      }|d   }	|	|d}
nt        || j                        }
|	j
                  |
   }||kD  rWt        d       g|	j                  z  }t        d|      ||
<   |	t        |         }	|r|d   ||
      |d<   |r|d   ||
      |d<   n||k  r~dg|	j                  z  }d||z
  f||
<   |dn|}t        j                  |	||      }	|r!t        j                  |d   ||
   d      |d<   |r!t        j                  |d   ||
   d      |d<   |	|d<   t        |      dk(  r|d   S t        |      S )NF)return_indexreturn_inversereturn_countsrJ   rp   r   r  r   )r   r   )r  )
r   uniquer6   r   r   rZ   slicer  r  r   )r*   sortedr  r  r  rJ   r?  rT  outputr3  dimvalues_countr  r  fills                  r   r  r    s    YY	!%#F NmfAYF<C#D!&&1C||C($T{mfkk1G D>GCLE'N+F#BZ5r
"1Igcl3q	D 6;;.I!45IcN"*1
DVVFItDFVV2J	#r
 FF6!9inaPq	F1IFq(6!9;eFm;r   c                 D    t        |       } t        j                  | |      S r(   )r   r   dsplitrf  s     r   r  r    rh  r   )r   )r   r   )NF)NFNr(   )gh㈵>g:0yE>F)NNN)r  )NN)Nr   F)r   )r  r  r  N)r   r   r   )r   r  )2   TNr   )FF)r  TFNr   )r  T
   Nr   )NlinearF)r   NN)r  N)left)r   )Nr   N)Ng      ?r  )valid)r  N)TFFFNNN)numpyr   	keras.srcr   r   keras.src.backendr   r   keras.src.backend.commonr   &keras.src.backend.common.backend_utilsr   r	   keras.src.backend.numpy.corer   r   r"   r8   r?   rB   rF   rL   rQ   rX   r\   r_   rc   rf   rj   rq   rv   rx   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   r   r   r   r   r   r  r  r  r  r  r  r  r  r  r"  r'  r*  r-  r0  r3  r8  r:  r=  r@  rC  rE  rG  rI  rL  rO  rQ  rS  rV  rX  r\  r^  r`  rc  re  rj  rl  rn  rp  rr  rt  rx  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  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  r   r9  r;  r=  r?  rA  rC  rF  r   rJ  rL  rN  rQ  rS  rU  rX  r[  r]  r_  rc  rh  rj  rl  rn  rp  rs  ru  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   <module>r     s      $ / + D M :+K$+ ID
'
(
3
J3
4
4
(:FF4
-
52
1
1

.6
<> 
 $ %2
%	$2)M"10
'
'C&1(
)#
!
3
4($-

*'
5G


)(
*!
 IJ,"$$# /"!&D
  .D
B346E
G
6"
DDCG )(
5		A3",#(P $!
7
=
"34*
5
+*	M,
-
 #'$ H#"J".#$
&
EIA,

#"
(&<'93 		<<~-r   