jax.numpy.fabs

jax.numpy.fabs(x, /)

Compute the absolute values element-wise.

LAX-backend implementation of numpy.fabs().

Original docstring below.

This function returns the absolute values (positive magnitude) of the data in x. Complex values are not handled, use absolute to find the absolute values of complex data.

Parameters:

x (array_like) – The array of numbers for which the absolute values are required. If x is a scalar, the result y will also be a scalar.

Returns:

y – The absolute values of x, the returned values are always floats. This is a scalar if x is a scalar.

Return type:

ndarray or scalar