jax.lax.population_count#

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

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

Parameters:

x (ArrayLike)

Return type:

Array