o o h @sddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZde d efd d Zdee efd efd d Zde d efddZde d dfddZde d dfddZdS)Eq)Expr)Integer)BooleanAnd) MatrixExpr) ShapeError)UnionargsreturncGshtdd|D\}}tgddt|dd|ddDddt|dd|ddDRS)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) cs|]}|jVqdSNshape.0argru/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/matrices/expressions/_shape.py z"is_matadd_valid..cs|] \}}t||VqdSrrrijrrrr"Ncsrrrrrrrr#rziprr 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=sz"is_matmul_valid..csrrrrrrrr>rNrrrr!rrris_matmul_valid's(r&rcCst|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) )rr"r#)rrrr is_squareAsr'NcGsjtdd|D\}}tttdd|dkrtd|tttdd|dkr3td|dS) z:Validate matrix shape for addition only for integer valuescsr rr)rxrrrrZrz*validate_matadd_integer..cSt|ttfSrr%intrr(rrr[z)validate_matadd_integer..rz!Matrices have mismatching shape: cSr)rr*r,rrrr-]r.N)r lensetfilterr r!rrrvalidate_matadd_integerXs r2cGsft|dd|ddD]#\}}|j|j}}t|ttfr0t|ttfr0||kr0td||q dS)z@Validate matrix shape for multiplication only for integer valuesNrrzMatrices are not aligned)r r#r"r%r+rr )r ABrrrrrvalidate_matmul_integeras "$ r5)sympy.core.relationalrsympy.core.exprrsympy.core.numbersrsympy.logic.boolalgrr"sympy.matrices.expressions.matexprrsympy.matrices.exceptionsr typingr r$r&r'r2r5rrrrs