jax.numpy.log2

jax.numpy.log2(x, /)

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

JAX implementation of numpy.log2.

参数:

x (ArrayLike) – 输入数组

返回:

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

返回类型:

Array

示例

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