o °2úhã @sØUdZddlZddlmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlmZdd lmZgd ¢Zd d d dddddœZd/dedededefdd„Zde jiZeeeefefed<zddlZejed<ejed<ejed<ejed<ejed<Wn eyŠYnwd0dedededefd d!„ZiZ eee!fed"<dede!fd#d$„Z"iZ#eee!fed%<dede!fd&d'„Z$e j%ej%ej%ej%e j%d(œZ&e j'ej'ej'ej'e j'd(œZ(d)d*„Z%d+d,„Z'dede!fd-d.„Z)dS)1z¥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)ÚAnyÚDictÚTuple)Úcupy)Újax)Ú object_arrays)Ú tensorflow)Útheano)Útorch)Úget_funcÚ has_einsumÚ has_tensordotÚbuild_expressionÚevaluate_constantsÚ has_backendz dask.arrayz theano.tensorzopt_einsum.backends.torchz jax.numpyzautograd.numpyz mars.tensor)Údaskr r rÚjaxlibÚautogradÚmarsÚfuncÚbackendÚdefaultÚreturncCsXzt t ||¡¡}|durt||ƒWSt|||ƒWSty+d}t| ||¡ƒ‚w)z½Try 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Ú_aliasesÚgetÚgetattrÚAttributeErrorÚformat)rrrÚlibÚ error_msg©r"úW/var/www/html/chatgem/venv/lib/python3.10/site-packages/opt_einsum/backends/dispatch.pyÚ _import_func%s" ÿúr$)ÚeinsumÚobjectÚ _cached_funcs)Ú tensordotÚnumpy)Ú transposer))r%r))r(r&)r*r&r)cCs>zt||fWStyt|||ƒ}|t||f<|YSw)ziReturn ``{backend}.{func}``, e.g. ``numpy.einsum``, or a default func if provided. Cache result. )r'ÚKeyErrorr$)rrrÚfnr"r"r#r Js   ýr Ú _has_einsumc CóXzt|WSty+z td|ƒdt|<Wn ty$dt|<Ynwt|YSw)zCCheck if ``{backend}.einsum`` exists, cache result for performance.r%TF)r-r+r r©rr"r"r#r Zó      ÿ ùr Ú_has_tensordotc Cr.)zFCheck if ``{backend}.tensordot`` exists, cache result for performance.r(TF)r1r+r rr/r"r"r#r kr0r )rr rr rcCót|||ƒS)zxBuild an expression, based on ``expr`` and initial arrays ``arrays``, that evaluates using backend ``backend``. )ÚCONVERT_BACKENDS©rÚarraysÚexprr"r"r#rŒórcCr2)zƒConvert constant arrays to the correct backend, and perform as much of the contraction of ``expr`` with these as possible. )ÚEVAL_CONSTS_BACKENDSr4r"r"r#r“r7rcCs | ¡tvS)zChecks if the backend is known.)Úlowerr3r/r"r"r#ršs r)N)r)N)*Ú__doc__rÚtypingrrrÚopt_einsum.backendsrÚ_cupyrÚ_jaxrrÚ _tensorflowr Ú_theanor Ú_torchÚ__all__rÚstrr$Ú object_einsumr'Ú__annotations__r)Únpr(r*r%ÚModuleNotFoundErrorr r-Úboolr r1r rr3rr8rr"r"r"r#Úsf       ù ÿ     ÿ û û