U ˜9%eFã@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úU/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/sympy/tensor/functions.pyÚ sz)TensorProduct.__new__..Úevaluateé)Úsympy.tensor.arrayr r r Z"sympy.matrices.expressions.matexprrZsympy.matrices.matricesrZsympy.strategiesrÚgetr rrÚ__new__rZOneÚ isinstancerÚappendÚlen)ÚclsÚargsÚkwargsr r r rrrrÚobjZarraysÚotherZscalarrZcoeffZnewargsrrrrs4         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 rr%rr(rÚ_get_args_shapes6s zTensorProduct._get_args_shapescCs| ¡}t|dƒS)Nr)r,Ú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)rrZshpr/rrr1Asÿz,TensorProduct.__getitem__..)ÚiterrZfromiterÚziprr,)r&r0rr/rr2?sþzTensorProduct.__getitem__N) Ú__name__Ú __module__Ú __qualname__Ú__doc__Z is_numberrr'r,Úpropertyr$r2rrrrr 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)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;rrrrÚs       < B