U -e @sddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZe edd d Zee efedd d Ze ed ddZe ddddZe ddddZdS)Eq)Expr)Integer)BooleanAnd) MatrixExpr) ShapeError)Union)argsreturncGsbtdd|D\}}tddt|dd|ddDddt|dd|ddDS)aReturn the symbolic condition how ``MatAdd``, ``HadamardProduct`` makes sense. Parameters ========== args The list of arguments of matrices to be tested for. Examples ======== >>> from sympy import MatrixSymbol, symbols >>> from sympy.matrices.expressions._shape import is_matadd_valid >>> m, n, p, q = symbols('m n p q') >>> A = MatrixSymbol('A', m, n) >>> B = MatrixSymbol('B', p, q) >>> is_matadd_valid(A, B) Eq(m, p) & Eq(n, q) css|] }|jVqdSNshape.0argrb/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/sympy/matrices/expressions/_shape.py sz"is_matadd_valid..css|]\}}t||VqdSr rrijrrrr"sNcss|]\}}t||VqdSr rrrrrr#sziprr rowscolsrrris_matadd_valid s ""r cGs>tdd|D\}}tddt|dd|ddDS)aReturn the symbolic condition how ``MatMul`` makes sense Parameters ========== args The list of arguments of matrices and scalar expressions to be tested for. Examples ======== >>> from sympy import MatrixSymbol, symbols >>> from sympy.matrices.expressions._shape import is_matmul_valid >>> m, n, p, q = symbols('m n p q') >>> A = MatrixSymbol('A', m, n) >>> B = MatrixSymbol('B', p, q) >>> is_matmul_valid(A, B) Eq(n, p) css|]}t|tr|jVqdSr ) isinstancerrrrrrr=s z"is_matmul_valid..css|]\}}t||VqdSr rrrrrr>sNrrrrrrris_matmul_valid'sr")rr cCst|j|jS)aReturn the symbolic condition how the matrix is assumed to be square Parameters ========== arg The matrix to be tested for. Examples ======== >>> from sympy import MatrixSymbol, symbols >>> from sympy.matrices.expressions._shape import is_square >>> m, n = symbols('m n') >>> A = MatrixSymbol('A', m, n) >>> is_square(A) Eq(m, n) )rrr)rrrr is_squareAsr#NcGsjtdd|D\}}tttdd|dkr>td|tttdd|dkrftd|dS) z:Validate matrix shape for addition only for integer valuescss|] }|jVqdSr r)rxrrrrZsz*validate_matadd_integer..cSst|ttfSr r!intrr$rrr[z)validate_matadd_integer..rz!Matrices have mismatching shape: cSst|ttfSr r%r'rrrr(]r)N)rlensetfilterr rrrrvalidate_matadd_integerXs r-cGsft|dd|ddD]F\}}|j|j}}t|ttfrt|ttfr||krtd||qdS)z@Validate matrix shape for multiplication only for integer valuesNrrzMatrices are not aligned)rrrr!r&rr )r ABrrrrrvalidate_matmul_integeras"$r0)Zsympy.core.relationalrZsympy.core.exprrZsympy.core.numbersrZsympy.logic.boolalgrrZ"sympy.matrices.expressions.matexprrZsympy.matrices.commonr typingr r r"r#r-r0rrrrs