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