jax.numpy.copysign

jax.numpy.copysign(x1, x2, /)

Change the sign of x1 to that of x2, element-wise.

LAX-backend implementation of numpy.copysign().

Original docstring below.

If x2 is a scalar, its sign will be copied to all elements of x1.

Parameters:

• x1 (array_like) – Values to change the sign of.

• x2 (array_like) – The sign of x2 is copied to x1. If x1.shape != x2.shape, they must be broadcastable to a common shape (which becomes the shape of the output).

Returns:

out – The values of x1 with the sign of x2. This is a scalar if both x1 and x2 are scalars.

Return type:

ndarray or scalar