piel.utils.numerical
====================

.. py:module:: piel.utils.numerical


Functions
---------

.. autoapisummary::

   piel.utils.numerical.round_complex_array


Module Contents
---------------

.. py:function:: round_complex_array(array: piel.types.ArrayTypes, to_absolute: bool = False)

   Rounds the elements of a complex JAX numpy array to the nearest integer.

   Parameters:
   - array: A complex JAX numpy array.
   - absolute: A boolean that determines whether the complex numbers are rounded to the nearest integers in their absolute value.

   Returns:
   - A JAX numpy array with the complex elements rounded to the nearest integers.