jax.numpy.powerΒΆ
-
jax.numpy.power(x1, x2)[source]ΒΆ First array elements raised to powers from second array, element-wise.
LAX-backend implementation of
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. Note that an integer type raised to a negative integer power will raise a ValueError.
- 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