jax.numpy.float_power

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

计算 x 的元素级基数 y 指数。

JAX 实现的 numpy.float_power。

参数:

• x (ArrayLike) – 标量或数组。指定底数。

• y (ArrayLike) – 标量或数组。指定指数。x 和 y 的形状应相同或具有广播兼容性。

返回:

一个包含以 x 为底，y 为指数的幂的数组，并提升为非精确数据类型。

返回类型:

数组

另请参阅

• 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)