U Zʹj`6@sddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZmZdd lmZmZmZmZmZmZdd lmZd d ZGdddeZddZ ddZ!GdddeZ"dS))Counter)Mulsympify)Add) ExprBuilder)default_sort_key)log) MatrixExpr)validate_matadd_integer) ZeroMatrix OneMatrix)unpackflatten conditionexhaustrm_idsort)sympy_deprecation_warningcGs,|s tdt|dkr |dSt|S)au Return the elementwise (aka Hadamard) product of matrices. Examples ======== >>> from sympy import hadamard_product, MatrixSymbol >>> A = MatrixSymbol('A', 2, 3) >>> B = MatrixSymbol('B', 2, 3) >>> hadamard_product(A) A >>> hadamard_product(A, B) HadamardProduct(A, B) >>> hadamard_product(A, B)[0, 1] A[0, 1]*B[0, 1] z#Empty Hadamard product is undefinedr) TypeErrorlenHadamardProductdoit)Zmatricesrb/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/matrices/expressions/hadamard.pyhadamard_products  rcs`eZdZdZdZdddfdd Zedd Zd d Zd d Z ddZ ddZ ddZ Z S)ra( Elementwise product of matrix expressions Examples ======== Hadamard product for matrix symbols: >>> from sympy import hadamard_product, HadamardProduct, MatrixSymbol >>> A = MatrixSymbol('A', 5, 5) >>> B = MatrixSymbol('B', 5, 5) >>> isinstance(hadamard_product(A, B), HadamardProduct) True Notes ===== This is a symbolic object that simply stores its argument without evaluating it. To actually compute the product, use the function ``hadamard_product()`` or ``HadamardProduct.doit`` TFN)evaluatecheckcsttt|}t|dkr"tdtdd|DsGsz*HadamardProduct.__new__..z Mix of Matrix and Scalar symbolszjPassing check to HadamardProduct 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)deep) listmaprr ValueErrorallrrvalidatesuper__new__r)clsrrargsobj __class__rrr+As"  zHadamardProduct.__new__cCs |jdjSNr)r-shapeselfrrrr2XszHadamardProduct.shapec stfdd|jDS)Ncsg|]}|jfqSr)_entryr ijkwargsrr ]sz*HadamardProduct._entry..)rr-)r4r7r8r9rr6rr5\szHadamardProduct._entrycCs ddlm}ttt||jSNr) transpose)$sympy.matrices.expressions.transposer<rr%r&r-r4r<rrr_eval_transpose_s zHadamardProduct._eval_transposec s|jfdd|jD}ddlmddlm}fdd|jDrfdd|jD}|d dtDj|j}t |g|}t |S) Nc3s|]}|jfVqdSr)rr!r7)hintsrrr#dsz'HadamardProduct.doit..r MatrixBase)ImmutableMatrixcsg|]}t|r|qSr)rr@rBrrr:is z(HadamardProduct.doit..csg|]}|kr|qSrrr@)explicitrrr:kscSsg|]}t|qSr)rfromiterr@rrrr:ls) funcr-Zsympy.matrices.matrixbaserCZsympy.matrices.immutablerDzipZreshaper2r canonicalize)r4rAexprrD remainderZexpl_matr)rCrErArrcs  zHadamardProduct.doitcCsdg}t|j}tt|D]>}|d||||g||dd}|t|qt|SNr) r%r-rangerdiffappendrrrF)r4xZtermsr-r7Zfactorsrrr_eval_derivativess  ,z HadamardProduct._eval_derivativec s<ddlm}ddlm}ddlm}fddtjD}g}|D]}jd|}j|dd} j|} t| |} dd g} fd dt| D} | D]} | j | j }| j | j }t |t |t ||g| t ||ggf| }|jdjdj| _ d| _|jdjd j| _d| _|g| _ || qqD|S) Nr ArrayDiagonalArrayTensorProduct _make_matrixcsg|]\}}|r|qSr)has)r!r7r"rPrrr:s zAHadamardProduct._eval_derivative_matrix_lines..r)rcs"g|]\}}j|dkr|qSr)r2r!r8er3rrr:srZ)0sympy.tensor.array.expressions.array_expressionsrSrU"sympy.matrices.expressions.matexprrW enumerater-_eval_derivative_matrix_linesr_lines_first_line_index_second_line_indexr_first_pointer_parent_first_pointer_index_second_pointer_parent_second_pointer_indexrO)r4rPrSrUrWZ with_x_indlinesindZ left_argsZ right_argsdZhadamdiagonalr7l1l2subexprr)r4rPrrd{sF         z-HadamardProduct._eval_derivative_matrix_lines)__name__ __module__ __qualname____doc__Zis_HadamardProductr+propertyr2r5r?rrQrd __classcell__rrr/rr)s rcCstddt}t|}||}tddtdd}||}dd}tdd|}||}t|trt|j}g}|D],\}}|dkr| |qz| t ||qzt|}td dt t }||}t |}|S) aCanonicalize the Hadamard product ``x`` with mathematical properties. Examples ======== >>> from sympy import MatrixSymbol, HadamardProduct >>> from sympy import OneMatrix, ZeroMatrix >>> from sympy.matrices.expressions.hadamard import canonicalize >>> from sympy import init_printing >>> init_printing(use_unicode=False) >>> A = MatrixSymbol('A', 2, 2) >>> B = MatrixSymbol('B', 2, 2) >>> C = MatrixSymbol('C', 2, 2) Hadamard product associativity: >>> X = HadamardProduct(A, HadamardProduct(B, C)) >>> X A.*(B.*C) >>> canonicalize(X) A.*B.*C Hadamard product commutativity: >>> X = HadamardProduct(A, B) >>> Y = HadamardProduct(B, A) >>> X A.*B >>> Y B.*A >>> canonicalize(X) A.*B >>> canonicalize(Y) A.*B Hadamard product identity: >>> X = HadamardProduct(A, OneMatrix(2, 2)) >>> X A.*1 >>> canonicalize(X) A Absorbing element of Hadamard product: >>> X = HadamardProduct(A, ZeroMatrix(2, 2)) >>> X A.*0 >>> canonicalize(X) 0 Rewriting to Hadamard Power >>> X = HadamardProduct(A, A, A) >>> X A.*A.*A >>> canonicalize(X) .3 A Notes ===== As the Hadamard product is associative, nested products can be flattened. The Hadamard product is commutative so that factors can be sorted for canonical form. A matrix of only ones is an identity for Hadamard product, so every matrices of only ones can be removed. Any zero matrix will make the whole product a zero matrix. Duplicate elements can be collected and rewritten as HadamardPower References ========== .. [1] https://en.wikipedia.org/wiki/Hadamard_product_(matrices) cSs t|tSrrrrYrrrzcanonicalize..cSs t|tSrryrYrrrrzr{cSs t|tSr)rr rYrrrrzr{cSs&tdd|jDrt|jS|SdS)Ncss|]}t|tVqdSr)rr )r!crrrr# sz/canonicalize..absorb..)anyr-r r2rYrrrabsorb s zcanonicalize..absorbcSs t|tSrryrYrrrrzr{rcSs t|tSrryrYrrrrz#r{)rrrrrrrr-itemsrO HadamardPowerrrr )rPruleZfunr~ZtallyZnew_argbaseexprrrrIs@S    rIcCsBt|}t|}|dkr|S|js*||S|jr8tdt||S)Nrz#cannot raise expression to a matrix)r is_Matrixr'r)rrrrrhadamard_power-srcsdeZdZdZfddZeddZeddZedd Zd d Z d d Z ddZ ddZ Z S)ra Elementwise power of matrix expressions Parameters ========== base : scalar or matrix exp : scalar or matrix Notes ===== There are four definitions for the hadamard power which can be used. Let's consider `A, B` as `(m, n)` matrices, and `a, b` as scalars. Matrix raised to a scalar exponent: .. math:: A^{\circ b} = \begin{bmatrix} A_{0, 0}^b & A_{0, 1}^b & \cdots & A_{0, n-1}^b \\ A_{1, 0}^b & A_{1, 1}^b & \cdots & A_{1, n-1}^b \\ \vdots & \vdots & \ddots & \vdots \\ A_{m-1, 0}^b & A_{m-1, 1}^b & \cdots & A_{m-1, n-1}^b \end{bmatrix} Scalar raised to a matrix exponent: .. math:: a^{\circ B} = \begin{bmatrix} a^{B_{0, 0}} & a^{B_{0, 1}} & \cdots & a^{B_{0, n-1}} \\ a^{B_{1, 0}} & a^{B_{1, 1}} & \cdots & a^{B_{1, n-1}} \\ \vdots & \vdots & \ddots & \vdots \\ a^{B_{m-1, 0}} & a^{B_{m-1, 1}} & \cdots & a^{B_{m-1, n-1}} \end{bmatrix} Matrix raised to a matrix exponent: .. math:: A^{\circ B} = \begin{bmatrix} A_{0, 0}^{B_{0, 0}} & A_{0, 1}^{B_{0, 1}} & \cdots & A_{0, n-1}^{B_{0, n-1}} \\ A_{1, 0}^{B_{1, 0}} & A_{1, 1}^{B_{1, 1}} & \cdots & A_{1, n-1}^{B_{1, n-1}} \\ \vdots & \vdots & \ddots & \vdots \\ A_{m-1, 0}^{B_{m-1, 0}} & A_{m-1, 1}^{B_{m-1, 1}} & \cdots & A_{m-1, n-1}^{B_{m-1, n-1}} \end{bmatrix} Scalar raised to a scalar exponent: .. math:: a^{\circ b} = a^b csVt|}t|}|jr$|jr$||St|trBt|trBt||t|||}|Sr)r is_scalarrr r)r*r+)r,rrr.r/rrr+rs  zHadamardPower.__new__cCs |jdSr1_argsr3rrrrszHadamardPower.basecCs |jdSrLrr3rrrrszHadamardPower.expcCs|jjr|jjS|jjSr)rrr2rr3rrrr2szHadamardPower.shapecKsx|j}|j}|jr$|j||f|}n|jr0|}ntd||jrV|j||f|}n|jrb|}ntd|||S)Nz)The base {} must be a scalar or a matrix.z-The exponent {} must be a scalar or a matrix.)rrrr5rr'format)r4r7r8r9rrabrrrr5s"zHadamardPower._entrycCsddlm}t||j|jSr;)r=r<rrrr>rrrr?s zHadamardPower._eval_transposecCs:|j|}|jt}||}t|||j||Sr)rrNrZ applyfuncrr)r4rPZdexpZlogbaseZdlbaserrrrQs   zHadamardPower._eval_derivativec sddlm}ddlm}ddlm}j|}|D]}ddg}fddt|D}|j|j }|j|j } t |t |t ||gj t jj d t || ggf||jd } | jdjdj|_d|_d|_ | jdjd j|_d|_d|_ | g|_q4|S) NrrTrRrV)rrZr[cs$g|]\}}jj|dkr|qSr^)rr2r_r3rrr:sz?HadamardPower._eval_derivative_matrix_lines..r) validatorrZ)rarUrSrbrWrrdrcrerfrgrrr _validater-rhrirjrk) r4rPrUrSrWlrr7rorprqrrrr3rrds>           z+HadamardPower._eval_derivative_matrix_lines)rsrtrurvr+rwrrr2r5r?rQrdrxrrr/rr9s8     rN)# collectionsrZ sympy.corerrZsympy.core.addrZsympy.core.exprrZsympy.core.sortingrZ&sympy.functions.elementary.exponentialrrbr Z!sympy.matrices.expressions._shaper r)Z"sympy.matrices.expressions.specialr r Zsympy.strategiesr rrrrrZsympy.utilities.exceptionsrrrrIrrrrrrs         ~