OpfDdZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Zgd Zd d d ddddZddZejejejejejejdZddZiZdZiZdZe je jejejejdZe je jejejejdZdZdZdZ dS)z Handles dispatching array operations to the correct backend library, as well as converting arrays to backend formats and then potentially storing them as constants. N) object_arrays)cupy)jax) tensorflow)theano)torch)get_func has_einsum has_tensordotbuild_expressionevaluate_constants has_backendz dask.arrayz theano.tensorzopt_einsum.backends.torchz jax.numpyzautograd.numpyz mars.tensor)daskrr rautogradmarsc  tjt||}|t ||nt |||S#t $r&d}t |||wxYw)zTry and import ``{backend}.{func}``. If library is installed and func is found, return the func; otherwise if default is provided, return default; otherwise raise an error. Nz{} doesn't seem to provide the function {} - see https://optimized-einsum.readthedocs.io/en/latest/backends.html for details on which functions are required for which contractions.) importlib import_module_aliasesgetgetattrAttributeErrorformat)funcbackenddefaultlib error_msgs \/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/opt_einsum/backends/dispatch.py _import_funcr!s >%hll7G&D&DEE%,_wsD!!!'#tW:U:UU >>>[ Y--gt<<=== >s AA0B)) tensordotnumpy) transposer#)einsumr#)r"object)r$r&)r%r&r#c t||fS#t$r"t|||}|t||f<|cYSwxYw)ziReturn ``{backend}.{func}``, e.g. ``numpy.einsum``, or a default func if provided. Cache result. ) _cached_funcsKeyErrorr!)rrrfns r r r <sYT7]++  $ 1 1') dGm$ s )==c t|S#t$rF td|dt|<n#t$r dt|<YnwxYwt|cYSwxYw)zHCheck if ``{backend}.einsum`` exists, cache result for performance. r%TF) _has_einsumr)r rrs r r r Ls $7## $$$ ) Xw ' ' '#'K  ) ) )#(K  )7####$/  A5AA  A A  AAc t|S#t$rF td|dt|<n#t$r dt|<YnwxYwt|cYSwxYw)zKCheck if ``{backend}.tensordot`` exists, cache result for performance. r"TF)_has_tensordotr)r rr-s r r r ^s 'g&& ''' , [' * * *&*N7 # # , , ,&+N7 # # # ,g&&&&'r.)rrrr rc.t|||S)zxBuild an expression, based on ``expr`` and initial arrays ``arrays``, that evaluates using backend ``backend``. )CONVERT_BACKENDSrarraysexprs r r r s G $VT 2 22c.t|||S)zConvert constant arrays to the correct backend, and perform as much of the contraction of ``expr`` with these as possible. )EVAL_CONSTS_BACKENDSr3s r rrs  ( 6 66r6c8|tvS)z$Checks if the backend is known. )lowerr2r-s r rrs ==??. ..r6)N)r#N)!__doc__rr#rr_cupyr_jaxr _tensorflowr_theanor _torch__all__rr!r"r$r% object_einsumr(r r,r r0r r r2rr8rr6r rEs  '''''' n n n  (      > > > >&"O!O"_"_'5       $ $ $ ' ' '$.&  "  $  0(  $  &  " 333777/////r6