jax.lax.clz

jax.lax.clz(x)

逐元素计算前导零的数量。

此函数直接降级到 stablehlo.count_leading_zeros 操作。

参数:

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

返回:

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

返回类型:

Array

另请参阅

• jax.lax.population_count()：计算每个元素中设置的位数。