`i,ddlZddlZddlZdZddZdS)Nc*tj|jtjr5t |d}n'|d}tj|S)N)cupy issubdtypedtypecomplexfloatingabspowersumsqrt)xsqnorms s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/sparse/linalg/_norm.py_sparse_frobenius_normrsg qw 455"Qa$$&&!! 9V  c tjj|st d||dvrt |S|}|d}n]t|tsHd} t|}n#t$rt |wxYw||krt ||f}d}t|dkr|\}}| |cxkr|krnn| |cxkr|ksntd|d|j ||z||zkrtd |dkrt|d krt|d kr5t|| S|t"jkr5t|| S|d kr5t|| S|t"j kr5t|| S|dvrt |Stdt|d krd|\}| |cxkr|ksntd|d|j |t"jkrGt|| S|t"j krGt|| S|dkrf|dkt"j| t"jS|d kr5t|| S|dvrZt3jt|d| S |d zn#t$rtdwxYwt3jt||| d |z Std)aNorm of a cupy.scipy.spmatrix This function is able to return one of seven different sparse matrix norms, depending on the value of the ``ord`` parameter. Args: x (sparse matrix) : Input sparse matrix. ord (non-zero int, inf, -inf, 'fro', optional) : Order of the norm (see table under ``Notes``). inf means numpy's `inf` object. axis : (int, 2-tuple of ints, None, optional): If `axis` is an integer, it specifies the axis of `x` along which to compute the vector norms. If `axis` is a 2-tuple, it specifies the axes that hold 2-D matrices, and the matrix norms of these matrices are computed. If `axis` is None then either a vector norm (when `x` is 1-D) or a matrix norm (when `x` is 2-D) is returned. Returns: ndarray : 0-D or 1-D array or norm(s). .. seealso:: :func:`scipy.sparse.linalg.norm` z)input is not sparse. use cupy.linalg.normN)Nfrof)rz6'axis' must be None, an integer or a tuple of integersrz Invalid axis z for an array with shape zDuplicate axes given.r)axis)Nrrz Invalid norm order for matrices.r)rNzInvalid norm order for vectors.z&Improper number of dimensions to norm.)cupyxscipysparseissparse TypeErrorrtocsr isinstancetupleintlen ValueErrorshapeNotImplementedErrorr r maxnumpyinfmintoarrayravelastypefloat32int_rr r ) r ordrmsgint_axisndrow_axiscol_axisas rnormr6s, ;  & &q ) )GDFFF |111%a(((  A | e $ $F !4yyHH ! ! !C..  ! 8  C.. { B 4yyA~~!(x$$$$"$$$$$")=)=)=)=2)=)=)=)=*"ddAGG-.. . b=HrM ) )455 5 !88% % BYY% % AXXq66::8:,,0022 2 EI  q66::8:,,0022 2 BYYq66::8:,,0022 2 UYJ  q66::8:,,0022 2 & & &)!,, ,?@@ @ Ta q 2 *"ddAGG-.. . %)  q66::1:%%--//5577 7 UYJ  q66::1:%%--//5577 7 AXXF??5=115515==CCEELL   AXXq66::1:%%++-- - I  9SVV\\!__00a08899??AA A Da D D D !BCCC D:c!ffll3//333;;QWEEKKMM MABBBs:B B$$Q**R)NN)r'rcupyx.scipy.sparserrr6rrr9s\ `C`C`C`C`C`Cr