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)