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)