CalcBucketsLimitAndOffset【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品√Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能根据bucket_list索引从ivf_counts和ivf_offset中获取对应桶的计数和偏移量并根据total_limit限制各桶的计数总和输出限制后的计数和对应偏移量。参数说明参数名输入/输出/属性描述数据类型数据格式bucket_list输入1-D tensor桶索引列表值为ivf_counts和ivf_offset的下标。INT32NDivf_counts输入1-D tensor每个桶的计数值。INT32NDivf_offset输入1-D tensor每个桶的偏移量。INT32、INT64NDtotal_limit属性所有桶计数总和的上限值。INT-buckets_limit输出1-D tensor限制后的各桶计数值总和不超过total_limit。INT32NDbuckets_offset输出1-D tensor与bucket_list对应的偏移量。INT32、INT64ND约束说明bucket_list中的值必须在[0, ivf_counts元素个数)和[0, ivf_offset元素个数)范围内ivf_counts和ivf_offset的元素个数应大于等于bucket_list中的最大索引值total_limit必须为非负整数调用说明调用方式样例代码说明图模式接口test_geir_calc_buckets_limit_and_offset通过算子IR接口方式调用CalcBucketsLimitAndOffset算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考