U ;qLeFã@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 ¡} | slt j|f|žŽ} | 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|f|ž|Ž} || ƒS) Nr)Ú NDimArrayÚ tensorproductÚArray)Ú MatrixExpr)Ú MatrixBase)ÚflattencSsg|] }t|ƒ‘qS©r)Ú.0Úargrrúm/home/p21-0144/sympy/latex2sympy2solve-back-end/sympyEq/lib/python3.8/site-packages/sympy/tensor/functions.pyÚ sz)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ÚobjZarraysÚotherZscalarrÚ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Úir-rrr8sz2TensorProduct._get_args_shapes..)rr r"r*rr-rÚ_get_args_shapes6s zTensorProduct._get_args_shapescCs| ¡}t|dƒS)Nr)r1Úsum)r+Z shape_listrrrr):szTensorProduct.shapecs,tˆƒ‰t ‡fdd„t|j| ¡ƒDƒ¡S)Nc3s.|]&\}}| t‡fdd„|Dƒƒ¡VqdS)c3s|]}tˆƒVqdSr()Únextr/©ÚindexrrÚ Bsz6TensorProduct.__getitem__...N)Ú __getitem__Útuple)rrZshpr4rrr6Asÿz,TensorProduct.__getitem__..)ÚiterrÚfromiterÚzipr"r1)r+r5rr4rr7?sþzTensorProduct.__getitem__N) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú is_numberrr,r1Úpropertyr)r7rrrrr 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)r.r)Ú NoShapeError)Úexprrrrr)Gs = ÿr)c@seZdZdZdS)rBan 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)r<r=r>r?rrrrrBŠsrBN)Úcollections.abcrÚ functoolsrÚsympy.core.exprrZsympy.core.mulrÚsympy.core.singletonrÚsympy.core.sympifyrÚsympy.core.parametersr r r)Ú ExceptionrBrrrrÚs       < B