jax.numpy.exp

jax.numpy.exp(x, /)

计算输入的逐元素指数。

JAX implementation of numpy.exp.

参数:

x (ArrayLike) – 输入数组或标量

返回:

包含 x 中每个元素的指数的数组，提升为不精确的数据类型。

返回类型:

Array

参见

• jax.numpy.log(): 计算输入的逐元素对数。

• jax.numpy.expm1(): 计算输入中每个元素的 \(e^x-1\)。

• jax.numpy.exp2(): 计算输入中每个元素的以2为底的指数。

示例

jnp.exp 遵循指数的性质，例如 \(e^{(a+b)} = e^a * e^b\)。

>>> x1 = jnp.array([2, 4, 3, 1])
>>> x2 = jnp.array([1, 3, 2, 3])
>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jnp.exp(x1+x2))
[  20.09 1096.63  148.41   54.6 ]
>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jnp.exp(x1)*jnp.exp(x2))
[  20.09 1096.63  148.41   54.6 ]

此属性对于复杂输入也适用：

>>> jnp.allclose(jnp.exp(3-4j), jnp.exp(3)*jnp.exp(-4j))
Array(True, dtype=bool)