`i(ddlZddlZddlmZddlmZddlmZddlZdZejddddZ ejd d d dZ dd Z dZ ddZ dZddZdS)N)_core)_decomposition)_utilcxtj|||fd}|tj|dd}|S)N)F compute_uvraxis)cupymoveaxisrsvd)xrow_axiscol_axisopyresults f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/linalg/_norms.py_multi_svd_normr sC a(H-x88A R"1777b A A AF M _norm_ord2)z?->lzb->lzB->Lzh->lzH->Lzi->lzI->Lzl->lzL->Lzq->qzQ->Q)ze->e)NNNfloatzf->fzd->d)z in0 * in0a + bout0 = sqrt(type_out0_raw(a))N_norm_ord2_complex)zF->fzD->d)z1in0.real() * in0.real() + in0.imag() * in0.imag()rrNFc t|jjtjs|t }||j}||dkr|dks |dkr|dvr|jjdkrMt| }||z}tj | }n)tj ||z }|r|d|z}|S|j}|tt!|}nFt#|ts1 t%|}n#t&$rt)dwxYw|f}t+|dkr|tjkr$t|||S|tj kr$t|||S|d kr8|d k|jj ||S|dkr$t| ||S||dkr4|jjdkrt5|||St7|||S t |n#t($rt9d wxYwt|}||z}| ||}|tj||j z}|St+|dkr|\} } | d kr| |z } | d kr| |z } d | cxkr|krnnd | cxkr|ksnt9d |d |j| | krt9d|dkr4t?j tj!d } tE|| | | }n|dkr4t?j tj!d} tE|| | | }n|dkrD| | kr| dz} t| | | }n[|tjkrD| | kr| dz} t| | | }n|dkrC| | kr| dz} t| | | }n|tj krC| | kr| dz} t| | | }nj|dvr4|jjdkrt5||}nDt7||}n2|dkrtE|| | tj }nt9d|r?tG|j} d| |d <d| |d<|| }|St9d)aReturns one of matrix norms specified by ``ord`` parameter. See numpy.linalg.norm for more detail. Args: x (cupy.ndarray): Array to take norm. If ``axis`` is None, ``x`` must be 1-D or 2-D. ord (non-zero int, inf, -inf, 'fro'): Norm type. axis (int, 2-tuple of ints, None): 1-D or 2-D norm is cumputed over ``axis``. keepdims (bool): If this is set ``True``, the axes which are normed over are left. Returns: cupy.ndarray N)ffroc)rz6'axis' must be None, an integer or a tuple of integersr keepdimsrzInvalid norm order for vectors.)dtypez Invalid axis z for an array with shape zDuplicate axes given.)indicesrrr )Nr"r!nucz Invalid norm order for matrices.z&Improper number of dimensions to norm.)$ issubclassr&typenumpyinexactastyperndimkindabsravelr sqrtsumreshapetuplerange isinstanceint Exception TypeErrorleninfmaxminrealrr ValueError reciprocalshape functoolspartialtakerlist)rordr r%r.sretndabsxrrop_maxop_min ret_shapes rnormrOs"$ aglEM 2 2 HHUOO |v KDAII#((sl22w|s"" NNQi((iQ .. /kk$+..J B |U2YY e $ $ Lt99DD L L LJLL L Lw 4yyA~~ %)  q66::4(:;; ; UYJ  q66::4(:;; ; AXXF??16<0044H5.. . AXXq66::4(:;; ; [C1HHw|s"")!$JJJJadX>>> > Dc  D D D !BCCC Dq66D SLD((x(88C DOCsy999 9CJ Ta!( a<< NH a<< NHX"""""""""qH'9'9'9'9r'9'9'9'9*"ddAGG-.. . x  455 5 !88&ty!<<>CC EI  (""A a&&**(*++//X/>>CC BYY(""A a&&**(*++//X/>>CC UYJ  (""A a&&**(*++//X/>>CC & & &w|s""(666 ... E\\!!XxBBCC?@@ @  )QW I!"Id1g !"Id1g ++i((C ABBBs>EE(J J:cTt|\}}|tj|zS)aReturns the determinant of an array. Args: a (cupy.ndarray): The input matrix with dimension ``(..., N, N)``. Returns: cupy.ndarray: Determinant of ``a``. Its shape is ``a.shape[:-2]``. .. seealso:: :func:`numpy.linalg.det` )slogdetr exp)asignlogdets rdetrVs(1::LD& $(6"" ""rc|jdkr0|dktSt j|d}|S|ddt |jd dztj |j j z}||k dtj S) aBReturn matrix rank of array using SVD method Args: M (cupy.ndarray): Input array. Its `ndim` must be less than or equal to 2. tol (None or float): Threshold of singular value of `M`. When `tol` is `None`, and `eps` is the epsilon value for datatype of `M`, then `tol` is set to `S.max() * max(M.shape) * eps`, where `S` is the singular value of `M`. It obeys :func:`numpy.linalg.matrix_rank`. Returns: cupy.ndarray: Rank of `M`. .. seealso:: :func:`numpy.linalg.matrix_rank` r rFr NrTr$r)r r&)r.anyr-r8rrr=rBr+finfor&epsr3intp)MtolSs r matrix_rankr_s" vzzQ||~~$$S)))1///A {uu"tu,,s17233>uEE 8\..v66??GGNN rrc4||||||S)aReturns the sum along the diagonals of an array. It computes the sum along the diagonals at ``axis1`` and ``axis2``. Args: a (cupy.ndarray): Array to take trace. offset (int): Index of diagonals. Zero indicates the main diagonal, a positive value an upper diagonal, and a negative value a lower diagonal. axis1 (int): The first axis along which the trace is taken. axis2 (int): The second axis along which the trace is taken. dtype: Data type specifier of the output. out (cupy.ndarray): Output array. Returns: cupy.ndarray: The trace of ``a`` along axes ``(axis1, axis2)``. .. seealso:: :func:`numpy.trace` )trace)rSoffsetrarbr&outs rr~r~s, 7765% 4 44r)NNF)N)rrrNN)r+r r cupy.linalgrrrCrcreate_reduction_funcrrrOrVr_rQr~rrrs& &&&&&& )U (B1 FF 1U0567 99~C~C~C~CH # # #44444KKK\555555r