jax.numpy.exp

jax.numpy.exp(x, /)

计算输入的逐元素指数。

JAX 对 numpy.exp 的实现。

参数:

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)