o +< iÌ ã@sˆddlmZmZmZddlmZddlmZddlm Z ddl m Z m Z ddœdd „Z Gd d „d eƒZddœd d „ZGdd„deƒZdS)é)ÚAddÚArgumentIndexErrorÚFunction)ÚPow)ÚS)Úlog)ÚcosÚsinT©ÚevaluatecCstt||dtj |dS©Nr )rrrÚOne)Úxr ©rúT/home/app/PyTorch/.pytorch/lib/python3.10/site-packages/sympy/codegen/scipy_nodes.pyÚ_cosm1src@s6eZdZdZdZd dd„Zdd„Zdd„Zd d „Zd S) Úcosm1zâ Minus one plus cosine of x, i.e. cos(x) - 1. For use when x is close to zero. Helper class for use with e.g. scipy.special.cosm1 See: https://docs.scipy.org/doc/scipy/reference/generated/scipy.special.cosm1.html écCs|dkr t|jŽ St||ƒ‚)ú@ Returns the first derivative of this function. r)r Úargsr©ÚselfZargindexrrrÚfdiffs  z cosm1.fdiffcKst|ƒS©N)r)rrÚkwargsrrrÚ_eval_rewrite_as_cosszcosm1._eval_rewrite_as_coscOó| t¡j|i|¤ŽSr)ÚrewriterÚevalf©rrrrrrÚ _eval_evalf ózcosm1._eval_evalfcKs8|j\}t|jdi|¤Žƒ}|t|ddkr|St|ƒS©NFr r)rrÚsimplifyr)rrrÚ candidaterrrÚ_eval_simplify#s zcosm1._eval_simplifyN©r) Ú__name__Ú __module__Ú __qualname__Ú__doc__Únargsrrr r%rrrrr s   rcCstt|||dtj |dSr )rrrr )rÚyr rrrÚ_powm1,sr-c@s6eZdZdZdZd dd„Zdd„Zdd „Zd d „Zd S)Úpowm1zý Minus one plus x to the power of y, i.e. x**y - 1. For use when x is close to one or y is close to zero. Helper class for use with e.g. scipy.special.powm1 See: https://docs.scipy.org/doc/scipy/reference/generated/scipy.special.powm1.html ércCs\|dkrt|jd|jdƒ|jd|jdS|dkr)t|jdƒt|jŽSt||ƒ‚)rrrr/)rrrrrrrrr8s * z powm1.fdiffcKs t||ƒSr)r-)rrr,rrrrÚ_eval_rewrite_as_PowCs zpowm1._eval_rewrite_as_PowcOrr)rrrrrrrr Fr!zpowm1._eval_evalfcKsL|j\}}t|jdi|¤Ž|jdi|¤Žƒ}|t||ddkr!|St||ƒSr")rr-r#r.)rrrr,r$rrrr%Is " zpowm1._eval_simplifyNr&) r'r(r)r*r+rr0r r%rrrrr.0s   r.N)Zsympy.core.functionrrrZsympy.core.powerrZsympy.core.singletonrZ&sympy.functions.elementary.exponentialrZ(sympy.functions.elementary.trigonometricrr rrr-r.rrrrÚs