EdFddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z Gdd eZed ZGd d eZd S))Iterable)singledispatch)Expr)Mul)Ssympify)global_parameterscDeZdZdZdZdZdZdZedZ dZ dS) TensorProductz, Generic class for tensor products. FcFddlm}m}m}ddlm}ddlm}ddlm }d|D}| dtj } | stj|g|R} | Sg} g} tj} |D]e}t#|t$||fr| ||9t#||fr| |`| |z} f| || z}t)| dkr|S|dkr|g| z}n| }tj|g|Ri|} || S) Nr) NDimArray tensorproductArray) MatrixExpr) MatrixBase)flattenc,g|]}t|Sr).0args 6lib/python3.11/site-packages/sympy/tensor/functions.py z)TensorProduct.__new__..s--- ---evaluate)sympy.tensor.arrayrrr"sympy.matrices.expressions.matexprrsympy.matrices.matricesrsympy.strategiesrgetr rr__new__rOne isinstancerappendlen)clsargskwargsrrrrrrrobjarraysotherscalarrcoeffnewargss rr"zTensorProduct.__new__sFFFFFFFFFFAAAAAA666666,,,,,,-----::j*;*DEE ,s*T***CJ  C#*i@AA  eeCjj))))C*//  S!!!!# }}f-- u::? L A: goGGGl33333F33ws||rc*t|jSN)r&shape)selfs rrankzTensorProduct.rank3s4:rc6ddlmfd|jDS)Nr)rc^g|])}t|dr|jn|j*S)r2)hasattrr2)rirs rrz2TensorProduct._get_args_shapes..8s8VVVq71g..BEE!HHNVVVr)rrr()r3rs @r_get_args_shapeszTensorProduct._get_args_shapes6s1,,,,,,VVVVDIVVVVrcJ|}t|dS)Nr)r9sum)r3 shape_lists rr2zTensorProduct.shape:s#**,, :r"""rcttjfdt|j|DS)Nc3zK|]5\}}|tfd|DV6dS)c36K|]}tVdSr1)next)rr8indexs r z6TensorProduct.__getitem__...Bs)!;!;!$u++!;!;!;!;!;!;rN) __getitem__tuple)rrshprAs rrBz,TensorProduct.__getitem__..Asd  S OOE!;!;!;!;s!;!;!;;; < <      r)iterrfromiterzipr(r9)r3rAs `rrCzTensorProduct.__getitem__?s`U |     4+@+@+B+BCC      rN) __name__ __module__ __qualname____doc__ is_numberr"r4r9propertyr2rCrrrr r sI   DWWW##X#     rr cTt|dr|jStd|z)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. r2zA%s does not have shape, or its type is not registered to shape().)r7r2 NoShapeError)exprs rr2r2Gs<ztWz KdR T TTrceZdZdZdS)rPan 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)rIrJrKrLrrrrPrPs DrrPN)collections.abcr functoolsrsympy.core.exprrsympy.core.mulrsympy.core.singletonrsympy.core.sympifyr sympy.core.parametersr r r2 ExceptionrPrrrr[s$$$$$$$$$$$$ """"""&&&&&&3333339 9 9 9 9 D9 9 9 x?T?T?TD     9     r