jax_absolute¶

probly.representation.jax_functions.jax_absolute(x: JaxArrayLike, /) → jax.Array[source]¶

probly.representation.jax_functions.jax_absolute(x: JaxOperand, /) → object

Compute absolute values elementwise, mirroring jax.numpy.absolute.

Parameters:: x – Real or complex operand.
Returns:: The elementwise magnitudes, or the result supplied by a custom override.