piel.utils.numerical

piel.utils.numerical#

Functions#

round_complex_array(array[, to_absolute])

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

Module Contents#

round_complex_array(array: piel.types.ArrayTypes, to_absolute: bool = False)[source]#

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.

Contents