U L?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 } | slt j|f|} | Sg} g} tj} |D]D}t|t||fr| ||q~t||fr| |q~| |9} q~| || }t| dkr|S|d kr|g| }n| }t j|f||} || S) Nr) NDimArray tensorproductArray) MatrixExpr) MatrixBase)flattencSsg|] }t|qSr).0argrrH/var/www/html/venv/lib/python3.8/site-packages/sympy/tensor/functions.py sz)TensorProduct.__new__..evaluate)sympy.tensor.arrayr r r Z"sympy.matrices.expressions.matexprrZsympy.matrices.matrixbaserZsympy.strategiesrgetr rr__new__rZOne isinstancerappendlen)clsargskwargsr r r rrrrobjZarraysotherZscalarrZcoeffZnewargsrrrrs4         zTensorProduct.__new__cCs t|jSN)rshapeselfrrrrank3szTensorProduct.rankcs ddlmfdd|jDS)Nrr cs&g|]}t|dr|jn|jqS)r$)hasattrr$rir(rrr8sz2TensorProduct._get_args_shapes..)rr rr%rr(r_get_args_shapes6s zTensorProduct._get_args_shapescCs|}t|dS)Nr)r,sum)r&Z shape_listrrrr$:szTensorProduct.shapecs,ttfddt|j|DS)Nc3s.|]&\}}|tfdd|DVqdS)c3s|]}tVqdSr#)nextr*indexrr Bsz6TensorProduct.__getitem__...N) __getitem__tuple)rrZshpr/rrr1Asz,TensorProduct.__getitem__..)iterrZfromiterziprr,)r&r0rr/rr2?szTensorProduct.__getitem__N) __name__ __module__ __qualname____doc__Z is_numberrr'r,propertyr$r2rrrrr s" r cCs t|dr|jStd|dS)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().N)r)r$ NoShapeError)exprrrrr$Gs = r$c@seZdZdZdS)r;an 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)r6r7r8r9rrrrr;sr;N)collections.abcr functoolsrZsympy.core.exprrZsympy.core.mulrZsympy.core.singletonrZsympy.core.sympifyrZsympy.core.parametersr r r$ Exceptionr;rrrrs       < B