wg:dZddlmZdZdZdZGddZy) zx Provides functionality for multidimensional usage of scalar-functions. Read the vectorize docstring for more details. wrapscttr}dnttr}dfdtt S)z Returns a structure with the same dimension as the specified argument, where each basic element is replaced by the function f applied on it. All other arguments stay the same. TFcpt|drtt|Sr|<n|<iS)N__iter__)hasattrlistmap)xargsf f_reducedis_argkwargsns `/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/sympy/core/multidimensional.pyrz#apply_on_element..f_reducedsD 1j !Iq)* *Qq d%f% %) isinstanceintstrr r )r r rr structurerrs```` @@rapply_on_elementr sS!SG  As 1I && Iy) **rcZ|Dcgc]}t|dr t|n|c}Scc}w)zU Returns a copy of an iterable object (also copying all embedded iterables). r)r iter_copy)ris rrr(s,DM MaGAz2IaL 9 MM Ms (cPt|dr|jSt|S)zR Returns a copy of the given structure (numpy-array, list, iterable, ..). copy)rrr)rs rstructure_copyr/s&y&!~~ Y rceZdZdZdZdZy) vectorizea Generalizes a function taking scalars to accept multidimensional arguments. Examples ======== >>> from sympy import vectorize, diff, sin, symbols, Function >>> x, y, z = symbols('x y z') >>> f, g, h = list(map(Function, 'fgh')) >>> @vectorize(0) ... def vsin(x): ... return sin(x) >>> vsin([1, x, y]) [sin(1), sin(x), sin(y)] >>> @vectorize(0, 1) ... def vdiff(f, y): ... return diff(f, y) >>> vdiff([f(x, y, z), g(x, y, z), h(x, y, z)], [x, y, z]) [[Derivative(f(x, y, z), x), Derivative(f(x, y, z), y), Derivative(f(x, y, z), z)], [Derivative(g(x, y, z), x), Derivative(g(x, y, z), y), Derivative(g(x, y, z), z)], [Derivative(h(x, y, z), x), Derivative(h(x, y, z), y), Derivative(h(x, y, z), z)]] cb|D]#}t|ttfrtd||_y)a The given numbers and strings characterize the arguments that will be treated as data structures, where the decorated function will be applied to every single element. If no argument is given, everything is treated multidimensional. za is of invalid typeN)rrr TypeErrormdargs)selfr#as r__init__zvectorize.__init__Qs6 8Aa#s, 677 8 rc6tfdS)zx Returns a wrapper for the one-dimensional function that can handle multidimensional arguments. c jr j}n%tt||jz}t|}|D]}t |t r||k\r||}d}nt |t r ||}d}tdsFrt|}t|||<nt|||<t |||}|cS|i|S#t$rYwxYw)NTFr) r#rangelenkeysrrrKeyErrorrr rr) r rr# arglengthrentryrresultr r$wrappers rr0z#vectorize.__call__..wrapperbs{{s4y)FKKM9D I "a%I~  GE!F3'! &q #F5*-#Dz"0"7Q$25$9q -gtVQGF!M+ ",d%f% %$! !s?C C$#C$r)r$r r0s``@r__call__zvectorize.__call__]s# q &  &@rN)__name__ __module__ __qualname____doc__r&r1rrr r 8s0 &rr N)r5 functoolsrrrrr r6rrr8s+ +<N KKr