o o hH@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z Gdd d eZed d ZGd d d eZdS))Iterable)singledispatch)Expr)Mul)Ssympify)global_parametersc@s@eZdZdZdZddZddZddZed d Z d d Z d S) TensorProductz, Generic class for tensor products. FcOsddlm}m}m}ddlm}ddlm}ddlm }dd|D}| dt j } | s7t j|g|R} | Sg} g} tj} |D]"}t|t||frR| ||q@t||fr^| |q@| |9} q@| || }t| dkrq|S|d kr{|g| }n| }t j|g|Ri|} || S) Nr) NDimArray tensorproductArray) MatrixExpr) MatrixBase)flattencSsg|]}t|qSr).0argrrj/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/tensor/functions.py sz)TensorProduct.__new__..evaluate)sympy.tensor.arrayr r r "sympy.matrices.expressions.matexprrsympy.matrices.matrixbasersympy.strategiesrgetr rr__new__rOne isinstancerappendlen)clsargskwargsr r r rrrrobjarraysotherscalarrcoeffnewargsrrrrs4         zTensorProduct.__new__cCs t|jSN)r!shapeselfrrrrank3s zTensorProduct.rankcs ddlmfdd|jDS)Nrr cs&g|]}t|dr |jn|jqS)r,)hasattrr,rir0rrr8s&z2TensorProduct._get_args_shapes..)rr r#r-rr0r_get_args_shapes6s zTensorProduct._get_args_shapescCs|}t|dS)Nr)r4sum)r. shape_listrrrr,:s zTensorProduct.shapecs,ttfddt|j|DS)Nc3s0|]\}}|tfdd|DVqdS)c3s|]}tVqdSr+)nextr2indexrr Bsz6TensorProduct.__getitem__...N) __getitem__tuple)rrshpr8rrr:As  z,TensorProduct.__getitem__..)iterrfromiterzipr#r4)r.r9rr8rr;?szTensorProduct.__getitem__N) __name__ __module__ __qualname____doc__ is_numberrr/r4propertyr,r;rrrrr s"  r cCst|dr|jStd|)a Return the shape of the *expr* as a tuple. *expr* should represent suitable object such as matrix or array. Parameters ========== expr : SymPy object having ``MatrixKind`` or ``ArrayKind``. Raises ====== NoShapeError : Raised when object with wrong kind is passed. Examples ======== This function returns the shape of any object representing matrix or array. >>> from sympy import shape, Array, ImmutableDenseMatrix, Integral >>> from sympy.abc import x >>> A = Array([1, 2]) >>> shape(A) (2,) >>> shape(Integral(A, x)) (2,) >>> M = ImmutableDenseMatrix([1, 2]) >>> shape(M) (2, 1) >>> shape(Integral(M, x)) (2, 1) You can support new type by dispatching. >>> from sympy import Expr >>> class NewExpr(Expr): ... pass >>> @shape.register(NewExpr) ... def _(expr): ... return shape(expr.args[0]) >>> shape(NewExpr(M)) (2, 1) If unsuitable expression is passed, ``NoShapeError()`` will be raised. >>> shape(Integral(x, x)) Traceback (most recent call last): ... sympy.tensor.functions.NoShapeError: shape() called on non-array object: Integral(x, x) Notes ===== Array-like classes (such as ``Matrix`` or ``NDimArray``) has ``shape`` property which returns its shape, but it cannot be used for non-array classes containing array. This function returns the shape of any registered object representing array. r,zA%s does not have shape, or its type is not registered to shape().)r1r, NoShapeError)exprrrrr,Gs =r,c@seZdZdZdS)rGan Raised when ``shape()`` is called on non-array object. This error can be imported from ``sympy.tensor.functions``. Examples ======== >>> from sympy import shape >>> from sympy.abc import x >>> shape(x) Traceback (most recent call last): ... sympy.tensor.functions.NoShapeError: shape() called on non-array object: x N)rArBrCrDrrrrrGsrGN)collections.abcr functoolsrsympy.core.exprrsympy.core.mulrsympy.core.singletonrsympy.core.sympifyrsympy.core.parametersr r r, ExceptionrGrrrrs      < B