cutcutcodec.core.compilation.sympy_to_torch.printer_atom.c_atan

cutcutcodec.core.compilation.sympy_to_torch.printer_atom.c_atan(out: Symbol, indexing: defaultdict[Symbol, str], c_type: str, arg: Atom) → tuple[set[str], set[Symbol], list[str]][source]

C atan operation.

Examples

>>> from sympy.abc import x
>>> from sympy.functions.elementary.trigonometric import atan
>>> import numpy as np
>>> from cutcutcodec.core.compilation.sympy_to_torch.lambdify import _lambdify_c
>>> from cutcutcodec.core.compilation.sympy_to_torch.printer import _print_atomic, _printer
>>> func = _lambdify_c(_printer([(x, atan(x))], {}, {x}))
>>> func(np.array([np.nan]))
array([nan])
>>> func(np.array([-np.inf, -1.0, 0.0, 1.0, np.inf])) / np.pi
array([-0.5 , -0.25,  0.  ,  0.25,  0.5 ])
>>>