o g’×iWã@s®ddlZddlZddlZddlZddlZej d¡rSz*ddlZddl Z ddl m Z e   dd¡Ze  dd¡Ze  dd¡ZeeefZdZWndZYerUd d „ZdSdSdS) éNÚ functorchééÚaÚbTFcCs|tjjjdd„ƒ}t d¡}t d¡}|d||ƒjdksJ‚t tj j ¡|d||ƒWdƒdS1s7wYdS)NcSs|| |d¡S)Né)Úsum)ÚxÚyÚaxes©r úT/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/test/test_experimental.pyÚeinfuncsz"test_functorchdim..einfunc)rré)réza [b c], a ([b] d) -> d [c] a)rrrz#(a + 1) [b c], a ([b] d) -> d [c] a) ÚeinxÚ experimentalÚ functorchdimÚadaptÚnpÚzerosÚshapeÚpytestÚraisesÚerrorsÚ EinxError)rr r r r r Útest_functorchdims   "ÿr)ÚconftestÚnumpyrÚ importlibrrÚutilÚ find_specrÚtorchÚ functorch.dimÚdimÚftdimrr ÚDimrrÚ availablerr r r r Ús*       ñ