jax.numpy.log2

jax.numpy.log2(x, /)

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

numpy.log2 的 JAX 实现。

参数:

x (ArrayLike) – 输入数组

返回:

一个数组，包含 x 中每个元素以 2 为底的对数，提升为非精确 dtype。

返回类型:

数组

示例

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