jax.lax.clz

jax.lax.clz(x)

逐元素前导零计数。

此函数直接降低为 stablehlo.count_leading_zeros 操作。

参数:

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

返回值:

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

返回类型:

Array

另请参阅

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