CANN/ops-math Roll循环位移算子

CANN/ops-math Roll循环位移算子 Roll【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math贡献说明贡献者贡献方贡献算子贡献时间贡献内容boxw987个人开发者Roll2026/06Roll 算子适配开源仓支持的产品型号Atlas A2 训练系列产品算子描述功能描述Roll沿给定维度对输入张量执行循环位移当dims为空时先按逻辑视图展平再执行一维循环位移最后按原始形状输出。原型信息算子类型(OpType)RollnameTypedata typeformat算子输入xtensoruint8, int8, bfloat16, float16, float32, int32, uint32ND算子输出ytensor与 x 相同ND属性shiftslistInt整型列表-dimslistInt整型列表-核函数名roll约束与限制仅支持ND格式。支持 0 维到 8 维输入。dims为空时shifts长度必须为 1。dims非空时shifts与dims长度必须一致。dims取值范围为[-rank, rank)。算子使用使用该算子前请参考社区版 CANN 开发套件包安装文档完成开发运行环境部署。编译部署cd ${git_clone_path}/ops-math bash build.sh --pkg --experimental --socascend910b --opsroll ./build_out/cann-ops-vendor_name-linux.arch.run调用说明调用方式调用样例说明aclnn 调用test_aclnn_roll.cpp通过 aclnnRoll 接口方式调用 Roll 算子【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考