jax.numpy.exp

jax.numpy.exp(x, /)

计算输入的逐元素指数。

numpy.exp 的 JAX 实现。

参数：

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

返回值：

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

返回类型：

数组

另请参阅

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