U Zʹj< @sddlmZmZddlmZddlmZmZmZddl m Z m Z ddl m Z mZddlmZddlmZddlmZmZmZmZmZmZmZdd lmZdd lmZdd lm Z dd l!m"Z#d dl$m%Z%d dl&m'Z'd dl(m)Z)d dl*m*Z*d dl+m,Z,d dl-m.Z.m/Z/m0Z0m1Z1Gddde'e Z2e 3e e2fe2ddZ4ddZ5ddZ6ddZ7ddZ8d d!Z9d"d#Z:d$d%Z;d&d'Zd*d+Z?d,d-Z@e@ed<d.S)/)askQ) handlers_dict)BasicsympifyS)mulMul)NumberIntegerDummyadjoint)rm_idunpacktypedflattenexhaustdo_onenew)NonInvertibleMatrixError) MatrixBase)sympy_deprecation_warning)validate_matmul_integer)Inverse) MatrixExpr)MatPow transpose)PermutationMatrix) ZeroMatrixIdentityGenericIdentity OneMatrixcseZdZdZdZeZddddddZedd Z e d d Z d$d d Z ddZ ddZfddZddZddZddZddZddZddZd%d d!Zd"d#ZZS)&MatMula A product of matrix expressions Examples ======== >>> from sympy import MatMul, MatrixSymbol >>> A = MatrixSymbol('A', 5, 4) >>> B = MatrixSymbol('B', 4, 3) >>> C = MatrixSymbol('C', 3, 6) >>> MatMul(A, B, C) A*B*C TFN)evaluatecheck_sympifycs|s jSttfdd|}|r2ttt|}tjf|}|\}}|dk rdtdddd|dk rtt ||s||S|r |S|S)Ncs j|kSN)identity)icls`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/matrices/expressions/matmul.py0z MatMul.__new__..zaPassing check to MatMul is deprecated and the check argument will be removed in a future version.z1.11z,remove-check-argument-from-matrix-operations)Zdeprecated_since_versionZactive_deprecations_targetF) r+listfiltermaprr__new__as_coeff_matricesrvalidate _evaluate)r.r'r(r)argsobjfactormatricesr/r-r0r6*s(  zMatMul.__new__cCst|Sr*) canonicalize)r.exprr/r/r0r9JszMatMul._evaluatecCs$dd|jD}|dj|djfS)NcSsg|]}|jr|qSr/ is_Matrix.0argr/r/r0 Psz MatMul.shape..r)r:rowscols)selfr=r/r/r0shapeNsz MatMul.shapec  snddlm}ddlm|\}}t|dkrD||d||fSdgt|ddgt|d}|d<|d<dd} |d| tdt|D]}t|<qt |ddD]\}} | j dd||<qfd d t |D}t |} t fd d |Drd }||| ftdddgt||} t dd |Ds\d}|rj| S| S)Nr)SumImmutableMatrixrrFcss d}td|V|d7}qdS)Nrzi_%ir )counterr/r/r0fbszMatMul._entry..fdummy_generatorcs,g|]$\}}|j||ddqS)r)rP)_entryrCr,rD)rPindicesr/r0rEosz!MatMul._entry..c3s|]}|VqdSr*hasrCvrLr/r0 qsz MatMul._entry..Tcss|]}t|ttfVqdSr*) isinstancer intrVr/r/r0rXysF)Zsympy.concrete.summationsrKZsympy.matrices.immutablerMr7lengetrangenext enumeraterJr fromiteranyzipdoit) rIr,jexpandkwargsrKcoeffr=Z ind_rangesrOrDZ expr_in_sumresultr/)rMrPrSr0rQSs6     z MatMul._entrycCsBdd|jD}dd|jD}t|}|jdkr:td||fS)NcSsg|]}|js|qSr/r@rCxr/r/r0rE~sz,MatMul.as_coeff_matrices..cSsg|]}|jr|qSr/r@rir/r/r0rEsFz3noncommutative scalars in MatMul are not supported.)r:r is_commutativeNotImplementedError)rIZscalarsr=rgr/r/r0r7}s  zMatMul.as_coeff_matricescCs|\}}|t|fSr*)r7r&rIrgr=r/r/r0 as_coeff_mmuls zMatMul.as_coeff_mmulc stt|jf|}||Sr*)superr&rer9)rIrfexpanded __class__r/r0resz MatMul.expandcCs2|\}}t|fdd|dddDS)aTransposition of matrix multiplication. Notes ===== The following rules are applied. Transposition for matrix multiplied with another matrix: `\left(A B\right)^{T} = B^{T} A^{T}` Transposition for matrix multiplied with scalar: `\left(c A\right)^{T} = c A^{T}` References ========== .. [1] https://en.wikipedia.org/wiki/Transpose cSsg|] }t|qSr/rrBr/r/r0rEsz*MatMul._eval_transpose..NrF)r7r&rcrmr/r/r0_eval_transposes  zMatMul._eval_transposecCs"tdd|jdddDS)NcSsg|] }t|qSr/rrBr/r/r0rEsz(MatMul._eval_adjoint..rF)r&r:rcrIr/r/r0 _eval_adjointszMatMul._eval_adjointcCs4|\}}|dkr0ddlm}|||SdS)Nr)trace)rnrvrc)rIr<mmulrvr/r/r0 _eval_traces  zMatMul._eval_tracecCs<ddlm}|\}}t|}||jttt||S)Nr) Determinant)Z&sympy.matrices.expressions.determinantryr7 only_squaresrGr r3r5)rIryr<r=Zsquare_matricesr/r/r0_eval_determinants  zMatMul._eval_determinantcCs>tdd|jDr6tdd|jdddDSt|S)Ncss|]}t|tr|jVqdSr*)rYr is_squarerBr/r/r0rXs z'MatMul._eval_inverse..css(|] }t|tr|n|dVqdS)rFN)rYrinverserBr/r/r0rXsrF)allr:r&rcrrtr/r/r0 _eval_inverses  zMatMul._eval_inversec s@dd}|r*tfdd|jD}n|j}tt|}|S)NdeepTc3s|]}|jfVqdSr*)rcrBhintsr/r0rXszMatMul.doit..)r\tupler:r>r&)rIrrr:r?r/rr0rcs   z MatMul.doitc sjddjD}ddjD}|rbt|}t|}|rb|rbt||krbtdfdd|D||gS)NcSsg|]}|jr|qSr/rkrir/r/r0rEsz#MatMul.args_cnc..cSsg|]}|js|qSr/rrir/r/r0rEsz"repeated commutative arguments: %scs$g|]}tj|dkr|qSr)r3r:count)rCcirtr/r0rEs)r:r[set ValueError)rIZcsetwarnrfZcoeff_cZcoeff_ncZclenr/rtr0args_cncszMatMul.args_cncc sddlmfddt|jD}g}|D]}|jd|}|j|dd}|r`t|}nt|jd}|rtfddt|D}nt|jd}|j| } | D]"} | || || | qq,|S)Nr Transposecsg|]\}}|r|qSr/rTrRrjr/r0rEs z8MatMul._eval_derivative_matrix_lines..cs"g|]}|jr|n|qSr/)rArc)rCr,rr/r0rEsr) r rr_r:r&r`r#rJreversed_eval_derivative_matrix_linesZ append_firstZ append_secondappend) rIrjZ with_x_indlinesindZ left_argsZ right_argsZ right_matZleft_revdr,r/)rrjr0rs$    z$MatMul._eval_derivative_matrix_lines)T)FT)__name__ __module__ __qualname____doc__Z is_MatMulr$r+r6 classmethodr9propertyrJrQr7rnrersrurxr{rrcrr __classcell__r/r/rqr0r&s(     *    r&cGs&|ddkr|dd}ttf|S)Nrr)rr&r:r/r/r0newmuls  rcCs>tdd|jDr:dd|jD}t|dj|djS|S)Ncss |]}|jp|jo|jVqdSr*)is_zerorAZ is_ZeroMatrixrBr/r/r0rXszany_zeros..cSsg|]}|jr|qSr/r@rBr/r/r0rEszany_zeros..rrF)rar:r"rGrH)rr=r/r/r0 any_zeross rcCstdd|jDs|Sg}|jd}|jddD]8}t|ttfr^t|ttfr^||}q4|||}q4||t|S)a Merge explicit MatrixBase arguments >>> from sympy import MatrixSymbol, Matrix, MatMul, pprint >>> from sympy.matrices.expressions.matmul import merge_explicit >>> A = MatrixSymbol('A', 2, 2) >>> B = Matrix([[1, 1], [1, 1]]) >>> C = Matrix([[1, 2], [3, 4]]) >>> X = MatMul(A, B, C) >>> pprint(X) [1 1] [1 2] A*[ ]*[ ] [1 1] [3 4] >>> pprint(merge_explicit(X)) [4 6] A*[ ] [4 6] >>> X = MatMul(B, A, C) >>> pprint(X) [1 1] [1 2] [ ]*A*[ ] [1 1] [3 4] >>> pprint(merge_explicit(X)) [1 1] [1 2] [ ]*A*[ ] [1 1] [3 4] css|]}t|tVqdSr*)rYrrBr/r/r0rXsz!merge_explicit..rrN)rar:rYrr rr&)matmulnewargslastrDr/r/r0merge_explicits    rcCs<|\}}tdd|}||kr4t|f|jS|SdS)z Remove Identities from a MatMul This is a modified version of sympy.strategies.rm_id. This is necesssary because MatMul may contain both MatrixExprs and Exprs as args. See Also ======== sympy.strategies.rm_id cSs |jdkS)NT)Z is_Identityrr/r/r0r16r2zremove_ids..N)rnrrr:)rr<rwrhr/r/r0 remove_ids's rcCs&|\}}|dkr"t|f|S|SNr)r7r)rr<r=r/r/r0factor_in_front<s rc Cs(|\}}|dg}tdt|D]}|d}||}t|trt|jtr|jj}t|}t||| dkr|d| t |j dg}q$t|trt|jtr|jj} t| }t| ||||krt |j d} | |d<t|||D] } | || <qq$|j dks&|j dkr2| |q$t|t rJ|j\} } n |tj} } t|t rn|j\}}n |tj}}| |kr| |}t | |jdd|d<q$nht|tsz |}Wntk rd}YnX|dk r| |kr| |}t | |jdd|d<q$| |q$t|f|S)aCombine consecutive powers with the same base into one, e.g. $$A \times A^2 \Rightarrow A^3$$ This also cancels out the possible matrix inverses using the knowledgebase of :class:`~.Inverse`, e.g., $$ Y \times X \times X^{-1} \Rightarrow Y $$ rrrFNF)r)r7r]r[rYrrDr&r:r3r#rJr|rrrZOnercrr}rr)rr<r:new_argsr,ABZBargslZAargsr+rdZA_baseZA_expZB_baseZB_expZnew_expZ B_base_invr/r/r0combine_powersBsX               rc Cs|j}t|}|dkr|S|dg}td|D]X}|d}||}t|tr|t|tr||jd}|jd}t|||d<q.||q.t|S)zGRefine products of permutation matrices as the products of cycles. rrrF)r:r[r]rYr!rr&) rr:rrhr,rrZcycle_1Zcycle_2r/r/r0combine_permutationss      rcCs|\}}|dg}|ddD]^}|d}t|trBt|tsN||q"||t|jd|jd||jd9}q"t|f|S)zj Combine products of OneMatrix e.g. OneMatrix(2, 3) * OneMatrix(3, 4) -> 3 * OneMatrix(2, 4) rrNrF)r7rYr%rpoprJr)rr<r:rrrr/r/r0combine_one_matricess   rcs|jtdkr~ddlm}djrNdjrN|fdddjDSdjr~djr~|fdddjDS|S)zr Simplify MatMul expressions but distributing rational term to MatMul. e.g. 2*(A+B) -> 2*A + 2*B rr)MatAddrcsg|]}t|dqSrr&rcrCZmatrr/r0rEsz$distribute_monom..csg|]}td|qS)rrrrr/r0rEs)r:r[ZmataddrZ is_MatAddZ is_Rational)rrr/rr0distribute_monoms  rcCs|dkSrr/rr/r/r0r1r2r1cGsp|dj|djkrtdg}d}t|D]>\}}|j||jkr,|t|||d|d}q,|S)z'factor matrices only if they are squarerrFz!Invalid matrices being multipliedr)rGrH RuntimeErrorr_rr&rc)r=outstartr,Mr/r/r0rzs rzcCsg}g}|jD] }|jr$||q||q|d}|ddD]h}||jkrrtt||rrt|jd}qD|| krtt ||rt|jd}qD|||}qD||t |S)z >>> from sympy import MatrixSymbol, Q, assuming, refine >>> X = MatrixSymbol('X', 2, 2) >>> expr = X * X.T >>> print(expr) X*X.T >>> with assuming(Q.orthogonal(X)): ... print(refine(expr)) I rrN) r:rArTrrZ orthogonalr#rJ conjugateZunitaryr&)r?Z assumptionsrZexprargsr:rrDr/r/r0 refine_MatMuls      rN)AZsympy.assumptions.askrrZsympy.assumptions.refinerZ sympy.corerrrZsympy.core.mulrr Zsympy.core.numbersr r Zsympy.core.symbolr Zsympy.functionsrZsympy.strategiesrrrrrrrZsympy.matrices.exceptionsrZsympy.matrices.matrixbaserZsympy.utilities.exceptionsrZ!sympy.matrices.expressions._shaperr8r}rZmatexprrZmatpowrr Z permutationr!specialr"r#r$r%r&Zregister_handlerclassrrrrrrrrrrulesr>rzrr/r/r/r0sZ   $         V*?  "