U Gv f@svdZgZddlmZmZmZmZddlmZm Z m Z dddddddddddddddddZ d d Z d d Z d dZdS)zrestructuredtext en) asanyarrayasarrayarrayzeros)aslinearoperatorLinearOperatorIdentityOperatorfdFD))r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r cCs$|dkr d}|dkrd}t||fS)NfdFDr ) _coerce_rules)xyrE/tmp/pip-unpacked-wheel-96ln3f52/scipy/sparse/linalg/_isolve/utils.pycoerces rcCs|SNrrrrridsrc Cs|}t|}|jd|jdkr2td|jf|jd}t|}|j|dfksx|j|fksxtd|jd|jd|jjdkr|d}d d }t|d r|jj}n||jj}t ||jj}t ||d }| }|d krNt|dr|j }nt }t|dr|j} nt } |t kr8| t kr8t|j|jd}nt|j|| |jd}nt|}|j|jkrltd|d krt||d } nvt|tr|dkr|} || } nLt||d } | j|dfks| j|fkstd|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 incompatibler r cSs|Srrrrrr postprocessQsz make_system..postprocessdtype)rNpsolverpsolve)shaper)matvecZrmatvecrz/matrix and preconditioner have different shapesZMbz and x0 )rr ValueErrorrrcharZastypehasattrrrrZravelrrrrrr isinstancestrcopyr) AMZx0bZA_NrZxtyperrrZbCopyrrr make_systemsV!               r(N)Z __docformat____all__ZnumpyrrrrZscipy.sparse.linalg._interfacerrrrrrr(rrrrs*