jax.numpy.float_power

jax.numpy.float_power(x, y, /)

計算元素級的底數 x 的 y 次方。

numpy.float_power 的 JAX 實作。

參數：

• x (ArrayLike) – 純量或陣列。指定底數。

• y (ArrayLike) – 純量或陣列。指定指數。x 和 y 應具有相同形狀或可廣播相容。

返回：

一個陣列，包含底數 x 的 y 次方，並提升為非精確的 dtype。

返回型別：

Array

另請參閱

• jax.numpy.exp()：計算輸入的元素級指數。

• jax.numpy.exp2()：計算輸入中每個元素的 2 次方指數。

範例

具有相同形狀的輸入

>>> x = jnp.array([3, 1, -5])
>>> y = jnp.array([2, 4, -1])
>>> jnp.float_power(x, y)
Array([ 9. ,  1. , -0.2], dtype=float32)

具有廣播相容性的輸入

>>> x1 = jnp.array([[2, -4, 1],
...                 [-1, 2, 3]])
>>> y1 = jnp.array([-2, 1, 4])
>>> jnp.float_power(x1, y1)
Array([[ 0.25, -4.  ,  1.  ],
       [ 1.  ,  2.  , 81.  ]], dtype=float32)

jnp.float_power 對於負值取非整數次方會產生 nan。

>>> jnp.float_power(-3, 1.7)
Array(nan, dtype=float32, weak_type=True)