jax.numpy.power

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

First array elements raised to powers from second array, element-wise.

LAX-backend implementation of numpy.power().

Original docstring below.

Raise each base in x1 to the positionally-corresponding power in x2. x1 and x2 must be broadcastable to the same shape.

An integer type raised to a negative integer power will raise a ValueError.

Negative values raised to a non-integral value will return nan. To get complex results, cast the input to complex, or specify the dtype to be complex (see the example below).

Parameters:

• x1 (array_like) – The bases.

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

Returns:

y – The bases in x1 raised to the exponents in x2. This is a scalar if both x1 and x2 are scalars.

Return type:

ndarray