o i@sddlZddlmZddlmZddlmZddlmZddl m Z Gddde Z d d Z d d Zd dZddZddZdddede deeBdBdejde ee dfBf ddZdS)N)Union)id)Backend)Tensorc@s eZdZdS)RemovedOperationErrorN)__name__ __module__ __qualname__r r R/home/ubuntu/.local/lib/python3.10/site-packages/einx/_src/frontend/removed_ops.pyr srcOtd)Nzeinx.vmap has been removed since version 0.4.0. Please use 'einx.{framework}.adapt_with_vmap' instead. See https://einx.readthedocs.io/en/latest/api/adapters.htmlrargskwargsr r r vmap rcOr )Nzeinx.vmap_with_axis has been removed since version 0.4.0. Please use one of the new adapters instead. See https://einx.readthedocs.io/en/latest/api/adapters.htmlrrr r r vmap_with_axisrrcOr )Nzeinx.reduce has been removed since version 0.4.0. Please use 'einx.{framework}.adapt_numpylike_reduce' instead. See https://einx.readthedocs.io/en/latest/api/adapters.htmlrrr r r reducerrcOr )Nzeinx.elementwise has been removed since version 0.4.0. Please use 'einx.{framework}.adapt_numpylike_elementwise' instead. See https://einx.readthedocs.io/en/latest/api/adapters.htmlrrr r r elementwiserrcOr )Nz\einx.arange has been removed since version 0.4.0. Please use einx.id with np.arange instead.rrr r r arange%sr)backend descriptiontensorsr parametersreturn.cOs*tjdtddt|g|Rd|i|S)Nzaeinx.rearrange is deprecated and will be removed in a future release. Please use einx.id instead.) stacklevelr)warningswarnDeprecationWarningr)rrrrr r r rearrange)sr")rtypingr numpy.typingnptopsrrrtypesr Exceptionrrrrrrstr ArrayLiketupler"r r r r s     >