jax.numpy.log2

jax.numpy.log2(x, /)

逐元素计算 x 的以 2 为底的对数。

JAX 实现的 numpy.log2。

参数：

x (ArrayLike) – 输入数组

返回值：

一个包含 x 中每个元素的以 2 为底的对数的数组，提升为不精确数据类型。

返回类型：

数组

示例

>>> x1 = jnp.array([0.25, 0.5, 1, 2, 4, 8])
>>> jnp.log2(x1)
Array([-2., -1.,  0.,  1.,  2.,  3.], dtype=float32)