jax.lax.population_count

jax.lax.population_count(x)

按元素统计 population count，计算每个元素中设置的位数。

此函数直接降低到 stablehlo.popcnt 操作。

参数:

x (ArrayLike) – 输入数组。必须具有整数 dtype。

返回:

一个与 x 具有相同形状和 dtype 的数组，其中包含输入中设置的位数。

返回类型:

数组

另请参阅

• jax.lax.clz(): 按元素计算前导零的个数。

• jax.numpy.bitwise_count(): 用于位计数的更灵活的 NumPy 风格 API。