o 8Vas@slddlmZddlmZddlmZmZmZmZddl m Z GdddeZ eddZ Gd d d e Zd S) )Iterable)singledispatch)ExprSMulsympify)global_parametersc@s@eZdZdZdZddZddZddZed d Z d d Z d S) TensorProductz, Generic class for tensor products. FcOsddlm}m}m}ddlm}m}ddlm}dd|D}| dt j } | s3t j |g|R} | Sg} g} tj} |D]"}t|t||frN| ||qsz)TensorProduct.__new__..evaluate)Zsympy.tensor.arrayr r r sympyr rZsympy.strategiesrgetrrr__new__rZOne isinstancerappendlen)clsargskwargsr r r r rrrobjZarraysotherZscalarrZcoeffZnewargsrrrrs2       zTensorProduct.__new__cCs t|jSN)rshapeselfrrrrank/s zTensorProduct.rankcs ddlmfdd|jDS)Nrr cs&g|]}t|dr |jn|jqS)r#)hasattrr#rir'rrr4s&z2TensorProduct._get_args_shapes..)rr rr$rr'r_get_args_shapes2s zTensorProduct._get_args_shapescCs|}t|dS)Nr)r+sum)r%Z shape_listrrrr#6s zTensorProduct.shapecs,ttfddt|j|DS)Nc3s0|]\}}|tfdd|DVqdS)c3s|]}tVqdSr")nextr)indexrr >sz6TensorProduct.__getitem__...N) __getitem__tuple)rrZshpr.rrr0=s  z,TensorProduct.__getitem__..)iterrZfromiterziprr+)r%r/rr.rr1;szTensorProduct.__getitem__N) __name__ __module__ __qualname____doc__Z is_numberrr&r+propertyr#r1rrrrr 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, Matrix, Integral >>> from sympy.abc import x >>> A = Array([1, 2]) >>> shape(A) (2,) >>> shape(Integral(A, x)) (2,) >>> M = Matrix([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().)r(r# NoShapeError)exprrrrr#Cs =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)r5r6r7r8rrrrr:sr:N)Zcollections.abcr functoolsrrrrrrZsympy.core.parametersrr r# Exceptionr:rrrrs  ; B