o +< iEã@sddlmZddlmZddlmZGdd„deƒZeƒZedd„ƒZdd d „Z d d „d d „e d fdd„Z ddd„Z dd„Z dd„Z dd„Zd S)é)Úcontextmanager)Úlocal©Ú expand_mulc@seZdZdd„ZdS)ÚDotProdSimpStatecCs d|_dS©N)Ústate)Úself©r úS/home/app/PyTorch/.pytorch/lib/python3.10/site-packages/sympy/matrices/utilities.pyÚ__init__s zDotProdSimpState.__init__N)Ú__name__Ú __module__Ú __qualname__r r r r r rs rccs*tj}z |t_dVW|t_dS|t_wr©Ú_dotprodsimp_stater)ÚxÚoldr r r Ú dotprodsimp s €rFcCsddlm}|||dS)z;Wrapper for simplify.dotprodsimp to avoid circular imports.r)r)Úwithsimp)Úsympy.simplify.simplifyr)ÚexprrZdpsr r r Ú _dotprodsimps  rcCs|Srr ©rr r r ÚsrNcCs0|dus tjdur |S|dustjdur|S|S)aSupport function for controlling intermediate simplification. Returns a simplification function according to the global setting of dotprodsimp operation. ``deffunc`` - Function to be used by default. ``offfunc`` - Function to be used if dotprodsimp has been turned off. ``onfunc`` - Function to be used if dotprodsimp has been turned on. ``dotprodsimp`` - True, False or None. Will be overridden by global _dotprodsimp_state.state if that is not None. FTr)ZdeffuncZofffuncZonfuncrr r r Ú_get_intermediate_simps  rcCst|dd|ƒS)zYSame as ``_get_intermediate_simp`` but returns bools instead of functions by default.FT)r)Údefaultrr r r Ú_get_intermediate_simp_bool3srcCs t|ddƒS)zReturns True if x is zero.Úis_zeroN)Úgetattrrr r r Ú_iszero:s r cCs t|ƒdkS)zNTests by expand_mul only, suitable for polynomials and rational functions.rrrr r r Ú_is_zero_after_expand_mul?s r!cCsddlm}||ƒS)z$ Wrapper to avoid circular imports. r)Úsimplify)rr")rr"r r r Ú _simplifyEs r#)F)FN)Ú contextlibrÚ threadingrZsympy.core.functionrrrrrrrr r!r#r r r r Ús     ÿ