o +< iã @sDdZddlmZmZmZddlmZddlmZddl m Z ddl m Z m Z mZGdd„dƒZed ed d ƒd d ZeƒZe d e ddgddggƒ¡e de ddgddggƒ¡e de ddgddggƒ¡e de ddgddggƒ¡e de ddgddggƒ¡e de ddgddggƒ¡e de de gedggƒ¡e de ddgdd ggƒ¡e de ddgdeggƒ¡e de ddgdeeedƒggƒ¡e dee ddgdd ggƒ¡e de ddgdd ggƒ¡e de gd¢gd¢gd¢gd¢gƒ¡e d ee ddgdd ggƒ¡e d!e edgde ggƒ¡d"S)#z7A cache for storing small matrices in multiple formats.é)ÚIÚRationalÚpi)ÚPow)Úexp)ÚMatrix)Úto_sympyÚto_numpyÚto_scipy_sparsec@sJeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dS)Ú MatrixCacheaA cache for small matrices in different formats. This class takes small matrices in the standard ``sympy.Matrix`` format, and then converts these to both ``numpy.matrix`` and ``scipy.sparse.csr_matrix`` matrices. These matrices are then stored for future recovery. ÚcomplexcCsi|_||_dS©N)Ú_cacheÚdtype)Úselfr©rú\/home/app/PyTorch/.pytorch/lib/python3.10/site-packages/sympy/physics/quantum/matrixcache.pyÚ__init__s zMatrixCache.__init__cCspz| ||¡Wn tyYnwz| ||¡Wn ty#Ynwz | ||¡WdSty7YdSw)zìCache a matrix by its name. Parameters ---------- name : str A descriptive name for the matrix, like "identity2". m : list of lists The raw matrix data as a SymPy Matrix. N)Ú _sympy_matrixÚ ImportErrorÚ _numpy_matrixÚ_scipy_sparse_matrix©rÚnameÚmrrrÚ cache_matrixs  ÿ ÿ ÿzMatrixCache.cache_matrixcCs,|j ||f¡}|dur|Std||fƒ‚)aGet a cached matrix by name and format. Parameters ---------- name : str A descriptive name for the matrix, like "identity2". format : str The format desired ('sympy', 'numpy', 'scipy.sparse') Nz3Matrix with name %s and format %s is not available.)rÚgetÚNotImplementedError©rrÚformatrrrrÚ get_matrix1s ÿÿzMatrixCache.get_matrixcCs||j||f<dSr )rrrrrÚ _store_matrixCszMatrixCache._store_matrixcCs| |dt|ƒ¡dS)NZsympy)r!rrrrrrFszMatrixCache._sympy_matrixcCó t||jd}| |d|¡dS)N©rÚnumpy)r rr!rrrrrIszMatrixCache._numpy_matrixcCr")Nr#z scipy.sparse)r rr!rrrrrMsz MatrixCache._scipy_sparse_matrixN)r ) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr r!rrrrrrrr s  r ééÿÿÿÿF)ÚevaluateZeye2éZop11Zop00Zop10Zop01ÚXÚYÚZÚSÚTéÚHZHsqrt2ZSWAP)r,rrr)rrr,r)rr,rr)rrrr,ZZXZZYN)r(Zsympy.core.numbersrrrZsympy.core.powerrZ&sympy.functions.elementary.exponentialrZsympy.matrices.denserZ!sympy.physics.quantum.matrixutilsrr r r Z sqrt2_invZ matrix_cacherrrrrÚs4   G(  ÿ "