jax.numpy.abs
jax.numpy.abs#
- jax.numpy.abs(x)#
Calculate the absolute value element-wise.
LAX-backend implementation of
numpy.absolute().Original docstring below.
np.absis a shorthand for this function.- Parameters
x (array_like) – Input array.
- Returns
absolute – An ndarray containing the absolute value of each element in x. For complex input,
a + ib, the absolute value is \(\sqrt{ a^2 + b^2 }\). This is a scalar if x is a scalar.- Return type
ndarray