cutcutcodec.core.compilation.sympy_to_torch.printer_atom.c_re¶

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

C real part.

Examples¶

>>> from sympy.abc import x
>>> from sympy import re
>>> 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, re(x))], {}, {x}))
>>> func(np.array([0.0, 1.0, -1.0]))
array([ 0.,  1., -1.])
>>> func(np.array([0.0j, 1.0j, -1.0j]))
array([ 0.+0.j,  0.+0.j, -0.+0.j])
>>>