PopulationCount

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：逐元素统计输入张量中每个 16 比特整数的二进制表示中 bit=1 的个数（population count，亦称 Hamming weight / popcount）。

• 计算公式：

$$y_i=\text{popcount}(x_i)$$

其中：

• $x_i$：输入张量的第 i 个元素，按无符号 16 位宽解释（INT16 负数按二进制补码逐比特计数）

• $y_i$：输出张量的第 i 个元素，取值范围为 $[0,16]$

• 使用场景：汉明距离计算、稀疏性分析、位掩码处理等需要统计二进制位数的场景。

参数说明

参数名	输入/输出	描述	数据类型	数据格式
x	输入	待统计的整数张量，对应公式中的 x。支持0-8维Tensor，0维表示标量。	INT16、UINT16	ND
y	输出	每个元素对应的位计数结果，对应公式中的 y。shape 与 x 一致。	UINT8	ND

约束说明

• 输入 x 的数据类型必须为 INT16 或 UINT16，输出 y 的数据类型固定为 UINT8。
• 输出 y 的 shape 必须与输入 x 的 shape 完全一致，不支持广播。
• 支持 0-8 维 Tensor，0 维表示标量。
• 支持空 Tensor（元素个数为 0），直接返回空结果。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_population_count	通过算子IR构图方式调用PopulationCount算子。