Bincount

产品支持情况

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

功能说明

  • 算子功能:计算非负整数数组中每个数的频率。minlength为输出tensor的最小size;当weights为空指针时,self中的self[i]每出现一次,则其频率加1,当weights不为空时,self[i]每出现一次,其频率加weights[i],最后存放到out的self[i]+1位置上;因此out大小为(self的最大值+1)与minlength中的最大值。

  • 计算公式:

    如果n是self在位置i上的值,如果指定了weights,则

    out[selfi]=out[selfi]+weightsiout[self_i] = out[self_i] + weights_i

    否则:

    out[selfi]=out[selfi]+1out[self_i] = out[self_i] + 1

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
array 输入 输入tensor,公式中的self_i。 INT32 ND
size 输入 指定输出tensor最小长度。 INT32 ND
weights 输入 输入tensor的权重,公式中的weights_i。 FLOAT ND
bins 输出 输出tensor,公式中的out_i。 FLOAT ND

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_bincount 通过aclnnBincount接口方式调用Bincount算子。