gdZgZddlmZmZmZmZddlmZm Z m Z idddddd d d d dd ddd dd dd dd dd dd dd dd dd dd Z dZ dZ dZdS)zrestructuredtext en) asanyarrayasarrayarrayzeros)aslinearoperatorLinearOperatorIdentityOperator)fr r )r dr )r Fr )r Dr )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r c8|dvrd}|dvrd}t||fS)NfdFDr ) _coerce_rules)xys W/granary/edev/quantum/lib/python3.11/site-packages/scipy/sparse/linalg/_isolve/utils.pycoercers-   1 c|SNrs ridrs Hrch|}t|}|jd|jdkrtd|jf|jd}t|}|j|dfks,|j|fks td|jd|jd|jjdvr|d}d }t|d r |jj}n||jj}t||jj}t|| }| }|t|d r|j }nt}t|dr|j} nt} |tur%| turt|j|j}nLt!|j|| |j}n.t|}|j|jkrtd|t#|| } nt%|t&r0|dkr)|} || } n^t+|| } | j|dfks,| j|fks td|jd| jd| } ||| ||fS)aMake a linear system Ax=b Parameters ---------- A : LinearOperator sparse or dense matrix (or any valid input to aslinearoperator) M : {LinearOperator, Nones} preconditioner sparse or dense matrix (or any valid input to aslinearoperator) x0 : {array_like, str, None} initial guess to iterative method. ``x0 = 'Mb'`` means using the nonzero initial guess ``M @ b``. Default is `None`, which means using the zero initial guess. b : array_like right hand side Returns ------- (A, M, x, b, postprocess) A : LinearOperator matrix of the linear system M : LinearOperator preconditioner x : rank 1 ndarray initial guess b : rank 1 ndarray right hand side postprocess : function converts the solution vector to the appropriate type and dimensions (e.g. (N,1) matrix) rz&expected square matrix, but got shape=z shapes of A z and b z are incompatiblerr c|Srrrs r postprocessz make_system..postprocessQsrdtype)rNpsolverpsolve)shaper)matvecrmatvecrz/matrix and preconditioner have different shapesMbz and x0 )rr" ValueErrorrrcharastypehasattrr#rrravelr rr!r rr isinstancestrcopyr) AMx0bA_Nrxtyper r!rbCopys r make_systemr6sB BAwqzQWQZN17*NNOOO  A1 A G!u  A4(((((()) ) w|6!! HHSMMq'  !& 5!', ' 'EA  A y 2h   YFFF 2i  jGGG R<>>AAqwvw%&W...AA Q   7ag  NOO O z !5 ! ! ! B    ::FFHHEA "E " " "Aq6!!QW__>AG>>#$7>>>?? ? GGII aA{ ""rN) __docformat____all__numpyrrrrscipy.sparse.linalg._interfacerrr rrrr6rrrr;sU% 433333333333 3 # y 3 )# /8 3 )# /8 3 !*# 09 3  !*#  09  3     `#`#`#`#`#r