jax.numpy.log2#

jax.numpy.log2(x,/)[source]#

Calculates the base-2 logarithm ofx element-wise.

JAX implementation ofnumpy.log2.

Parameters:

x (ArrayLike) – Input array

Returns:

An array containing the base-2 logarithm of each element inx, promotesto inexact dtype.

Return type:

Array

Examples

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