jax.lax.population_count

jax.lax.population_count#

jax.lax.population_count(x)[source]#

Elementwise popcount, count the number of set bits in each element.

Parameters:

x (jax.typing.ArrayLike)

Return type:

Array