a SG5dFã@s„ddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z Gdd „d eƒZed d „ƒZGd d „d eƒZdS)é)ÚIterable)Úsingledispatch)ÚExpr)ÚMul)ÚS©Úsympify)Úglobal_parametersc@s@eZdZdZdZdd„Zdd„Zdd„Zed d „ƒZ d d „Z d S)Ú TensorProductz, Generic class for tensor products. FcOsddlm}m}m}ddlm}ddlm}ddlm }dd„|Dƒ}|  dt j ¡} | snt j|g|¢RŽ} | Sg} g} tj} |D]D}t|t||fƒr¤|  ||ƒ¡q€t||fƒr¼|  |¡q€| |9} q€| || Ž}t| ƒdkrâ|S|d krö|g| }n| }t j|g|¢Ri|¤Ž} || ƒS) Nr)Ú NDimArrayÚ tensorproductÚArray)Ú MatrixExpr)Ú MatrixBase)ÚflattencSsg|] }t|ƒ‘qS©r)Ú.0ÚargrrúR/var/www/html/django/DPS/env/lib/python3.9/site-packages/sympy/tensor/functions.pyÚ óz)TensorProduct.__new__..Úevaluateé)Úsympy.tensor.arrayr r r Ú"sympy.matrices.expressions.matexprrÚsympy.matrices.matricesrZsympy.strategiesrÚgetr rrÚ__new__rÚOneÚ isinstancerÚappendÚlen)ÚclsÚargsÚkwargsr r r rrrrÚobjÚarraysÚotherÚscalarrÚcoeffÚnewargsrrrrs4         zTensorProduct.__new__cCs t|jƒS©N)r!Úshape©ÚselfrrrÚrank3szTensorProduct.rankcs ddlm‰‡fdd„|jDƒS)Nr©r cs&g|]}t|dƒr|jnˆ|ƒj‘qS)r,)Úhasattrr,©rÚir0rrr8rz2TensorProduct._get_args_shapes..)rr r#r-rr0rÚ_get_args_shapes6s zTensorProduct._get_args_shapescCs| ¡}t|dƒS)Nr)r4Úsum)r.Ú shape_listrrrr,:szTensorProduct.shapecs,tˆƒ‰t ‡fdd„t|j| ¡ƒDƒ¡S)Nc3s.|]&\}}| t‡fdd„|Dƒƒ¡VqdS)c3s|]}tˆƒVqdSr+)Únextr2©ÚindexrrÚ Brz6TensorProduct.__getitem__...N)Ú __getitem__Útuple)rrÚshpr8rrr:Asÿz,TensorProduct.__getitem__..)ÚiterrÚfromiterÚzipr#r4)r.r9rr8rr;?sþzTensorProduct.__getitem__N) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú is_numberrr/r4Úpropertyr,r;rrrrr s" r cCs t|dƒr|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)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)rArBrCrDrrrrrGŠsrGN)Úcollections.abcrÚ functoolsrÚsympy.core.exprrZsympy.core.mulrÚsympy.core.singletonrÚsympy.core.sympifyrÚsympy.core.parametersr r r,Ú ExceptionrGrrrrÚs       < B