(ph#XSSKJr SSKrSSKJr SSKJrJr SSK J r J r S/r S Sjr g) )IterableN)_asarray_validated) block_diag LinAlgError)_compute_lworkget_lapack_funcscossinc U(dU(aUcSO [U5nUcSO [U5n[USS9n[R"UR6(d[ SUR35eURSnX:dUS::a[ SUSURSS 35eX':dUS::a[ S US URSS 35eUSU2SU24USU2US24XS2SU24XS2US244uppGO&[ U[5(d [ S 5e[U5S :wa[ S[U535eSU5upp[/SQXX/5H&upU RSS:XdM[ U S35e URupU RupU RX4:wa[ SX4SU R35eU RX4:wa[ SX4SU R35eX-X/-:wa[ SX-SX/-S35eX-n[XX4Vs/sHn[R"U5PM sn5nU(aSOSn[UUS-/XX/5unn[UXqUS9nU(a USUSS.OSU0nU"S%XXUUUUSUS. UD6GtnnnnnnnURU-nUS:a[ S U*S!U35eUS:a[US"U35eU(a UU4UUU44$[!UU5n[!UU5n [R""[R$"U55n![R""[R&"U55n"[)XXq- Xr- 5n#[)X5U#- n$[)XU- 5U#- n%[)Xq- U5U#- n&[)Xq- Xr- 5U#- n'[R*"[R,"U$U%U&U'U#/5UR.S#9n([R0"Xw4UR.S#9n)U(SU$2SU$24U)SU$2SU$24'U$U#-n*U$U#-U%-n+U$U&-U'-S$U#--n,U$U&-U'-S$U#--U%-n-U(a U(SU%2SU%24O U(SU%2SU%24*U)U*U+2U,U-24'UU'-U#-n*UU'-U#-U&7-n+U$U#-n,U$U#-U&-n-U(a U(SU&2SU&24*O U(SU&2SU&24U)U*U+2U,U-24'U(SU'2SU'24U)XU'-2X"U'-24'U!U)U$U$U#-2U$U$U#-24'U!U)UU'-UU'-U#-2U$U#-U&-U'-S$U#-U$-U&-U'-24'U$n*U$U#-n+U$U&-U'-U#-n,U$U&-U'-S$U#--n-U(aU"OU"*U)U*U+2U,U-24'U(aU"*OU"U)UU'-UU'-U#-2U$U$U#-24'UU)U 4$s snf)&uo Compute the cosine-sine (CS) decomposition of an orthogonal/unitary matrix. X is an ``(m, m)`` orthogonal/unitary matrix, partitioned as the following where upper left block has the shape of ``(p, q)``:: ┌ ┐ │ I 0 0 │ 0 0 0 │ ┌ ┐ ┌ ┐│ 0 C 0 │ 0 -S 0 │┌ ┐* │ X11 │ X12 │ │ U1 │ ││ 0 0 0 │ 0 0 -I ││ V1 │ │ │ ────┼──── │ = │────┼────││─────────┼─────────││────┼────│ │ X21 │ X22 │ │ │ U2 ││ 0 0 0 │ I 0 0 ││ │ V2 │ └ ┘ └ ┘│ 0 S 0 │ 0 C 0 │└ ┘ │ 0 0 I │ 0 0 0 │ └ ┘ ``U1``, ``U2``, ``V1``, ``V2`` are square orthogonal/unitary matrices of dimensions ``(p,p)``, ``(m-p,m-p)``, ``(q,q)``, and ``(m-q,m-q)`` respectively, and ``C`` and ``S`` are ``(r, r)`` nonnegative diagonal matrices satisfying ``C^2 + S^2 = I`` where ``r = min(p, m-p, q, m-q)``. Moreover, the rank of the identity matrices are ``min(p, q) - r``, ``min(p, m - q) - r``, ``min(m - p, q) - r``, and ``min(m - p, m - q) - r`` respectively. X can be supplied either by itself and block specifications p, q or its subblocks in an iterable from which the shapes would be derived. See the examples below. Parameters ---------- X : array_like, iterable complex unitary or real orthogonal matrix to be decomposed, or iterable of subblocks ``X11``, ``X12``, ``X21``, ``X22``, when ``p``, ``q`` are omitted. p : int, optional Number of rows of the upper left block ``X11``, used only when X is given as an array. q : int, optional Number of columns of the upper left block ``X11``, used only when X is given as an array. separate : bool, optional if ``True``, the low level components are returned instead of the matrix factors, i.e. ``(u1,u2)``, ``theta``, ``(v1h,v2h)`` instead of ``u``, ``cs``, ``vh``. swap_sign : bool, optional if ``True``, the ``-S``, ``-I`` block will be the bottom left, otherwise (by default) they will be in the upper right block. compute_u : bool, optional if ``False``, ``u`` won't be computed and an empty array is returned. compute_vh : bool, optional if ``False``, ``vh`` won't be computed and an empty array is returned. Returns ------- u : ndarray When ``compute_u=True``, contains the block diagonal orthogonal/unitary matrix consisting of the blocks ``U1`` (``p`` x ``p``) and ``U2`` (``m-p`` x ``m-p``) orthogonal/unitary matrices. If ``separate=True``, this contains the tuple of ``(U1, U2)``. cs : ndarray The cosine-sine factor with the structure described above. If ``separate=True``, this contains the ``theta`` array containing the angles in radians. vh : ndarray When ``compute_vh=True`, contains the block diagonal orthogonal/unitary matrix consisting of the blocks ``V1H`` (``q`` x ``q``) and ``V2H`` (``m-q`` x ``m-q``) orthogonal/unitary matrices. If ``separate=True``, this contains the tuple of ``(V1H, V2H)``. References ---------- .. [1] Brian D. Sutton. Computing the complete CS decomposition. Numer. Algorithms, 50(1):33-65, 2009. Examples -------- >>> import numpy as np >>> from scipy.linalg import cossin >>> from scipy.stats import unitary_group >>> x = unitary_group.rvs(4) >>> u, cs, vdh = cossin(x, p=2, q=2) >>> np.allclose(x, u @ cs @ vdh) True Same can be entered via subblocks without the need of ``p`` and ``q``. Also let's skip the computation of ``u`` >>> ue, cs, vdh = cossin((x[:2, :2], x[:2, 2:], x[2:, :2], x[2:, 2:]), ... compute_u=False) >>> print(ue) [] >>> np.allclose(x, u @ cs @ vdh) True NrT) check_finitez=Cosine Sine decomposition only supports square matrices, got rz invalid p=z, 0cossin..s:1bmmA..s#%)x11x12x21x22z can't be emptyz Invalid x12 dimensions: desired z, got z Invalid x21 dimensions: desired zWThe subblocks have compatible sizes but don't form a square array (instead they form a rz5 array). This might be due to missing p, q arguments.uncsdorcsd_lwork)mpq)lworklrworkr F) rrrr compute_u1 compute_u2 compute_v1t compute_v2ttranssignszillegal value in argument z of internal z did not converge: )dtype)intrrequalshape ValueError isinstancerlenzipany iscomplexobjr rtypecoderrdiagcossinmineyemaxr(zeros).Xrrseparate swap_sign compute_u compute_vhrrrrrnameblockmmpmmqrcplxdrivercsd csd_lworkr lwork_args_thetau1u2v1hv2hinfo method_nameUVDHcsrn11n12n21n22IdCSxsxeysyes. rr r sF AAAAA qt 4xx!//0wwi9: : GGAJ 6Q!Vz!F1771:,jIJ J 6Q!Vz!F1771:,jIJ Jrr2A2vY"1"ab& 1R!V9aABiG#s 8 $ $:; ; q6Q;558VH>? ?;:#; ##35KD{{1~" D6!9::5yy99 99 ?zJ$$'II;01 1 99 ?zJ$$'II;01 1 7ag ""#'!AG95??@ @ G Ss,@A,@q",@A BDW'F%vv/@&A'*&:rhs)$/04 *(-7;Rrg