`i bddlZddlZddlZddlmZdZgdZdZdZej dZ dS)N)_utilc\tj|tj||jd|jdkrtd|dddtjddf|dtjddddfz}|d|jddzS)av Khatri-rao product A column-wise Kronecker product of two matrices Parameters ---------- a : (n, k) array_like Input array b : (m, k) array_like Input array Returns ------- c: (n*m, k) ndarray Khatri-rao product of `a` and `b`. See Also -------- .. seealso:: :func:`scipy.linalg.khatri_rao` z6The number of columns for both arrays should be equal..N))r _assert_2dshape ValueErrorcupynewaxisreshape)abcs p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/linalg/_matfuncs.py khatri_raors0 Q QwqzQWQZ,-- - #qqq$, !"QsDL!!!QQQ'>%??A 99UQWQRR[( ) )))gD`lCgD`\Cg`=Hb;Cg eCgJXBg"5Bg/cBg\L8BgpķAgsyAgS-Ag@gf@g?g{Gz@c |jdkrtjd|jS|jd}tj|jtjr|jn tj}tj| |z }|tj |||zz }tj |d }|tk}|rLtt!jt!jt'|tz dz}nd}|d|zz}||z}||z} || z} tj |jd|} tjt*|} t-| || | | \} }}}|| | z|zz}| |z|z}tj | |z||z}|}t1|D]}||z}tj|jtjrt5j|nt!j|}||z}|S)aCompute the matrix exponential. Parameters ---------- a : ndarray, 2D Returns ------- matrix exponential of `a` Notes ----- Uses (a simplified) version of Algorithm 2.3 of [1]_: a [13 / 13] Pade approximant with scaling and squaring. Simplifications: * we always use a [13/13] approximate * no matrix balancing References ---------- .. [1] N. Higham, SIAM J. MATRIX ANAL. APPL. Vol. 26(4), p. 1179 (2005) https://doi.org/10.1137/04061101X r)rr)dtyper)ordr)sizer zerosrr issubdtypeinexactfloat64diagsumeyelinalgnormitemth13intmathceillog2floatasarrayr _expm_innersolverangecomplexfloatingcmathexp)rna_dtypemuAnrmAscalesA2A4A6Ebbu1u2v1v2uvr13x_emus rexpmrE>s56 v{{z&0000  A 1agg$(L  1    a B DHQg & & &r ))A ;  A1  % % * * , ,D 4KE   $)E$KK$$67788 9 9A = AIA QB bB bB 7+++A aw ' ' 'B BB33NBB R"Wr\A R" A +  QBFAE * *C A 1XX E ? $&(( :%)B----1Xb\\HA HrcV|d|z|d|zz|d|zz}|d|z|d|zz|d|zz|d|zz}|d|z|d |zz|d |zz}|d |z|d |zz|d |zz|d|zz}||||fS)N r rr) r9r6r7r8rr;r<r=r>s rr)r)s 2rAbE"H qtBw &B 1b1Q47 QqT"W $qtAv -B 2rAbE"H qtBw &B 1b1Q47 QqT"W $qtAv -B r2r>r) r$r-r cupy.linalgrrrr"rEfuser)rRrrrUs  * * *H     K K K \ r