jax.lax.population_count¶

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

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

Parameters

x (Any) –

Return type

Any