CANN/ops-math SplitV算子

📅 2026/7/10 20:20:51 ✍️ 编辑团队 👁️ 阅读次数

$CANN/ops-math SplitV算子$

SplitV【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Atlas A2 训练系列产品/Atlas A2 推理系列产品√功能说明算子功能根据size_splits将张量沿维度split_dim拆分为num_split更小的张量。参数说明参数名输入/输出/属性描述数据类型数据格式x输入需要切分的tensor列表。BOOL、INT8、UINT8、BFLOAT16、FLOAT16、INT16、UINT16、FLOAT32、INT32、UINT32、INT64、UINT64NDsize_splits输入指定一个列表其中包含沿分割维度的每个输出张量的大小。INT64NDsplit_dim输入指定沿其分割的维度。INT64NDnum_split属性指定要分割的tensor个数。INT-y输出输出结果。BOOL、INT8、UINT8、BFLOAT16、FLOAT16、INT16、UINT16、FLOAT32、INT32、UINT32、INT64、UINT64ND约束说明size_splits中的每个元素都大于或等于1。size_splits的长度等于num_split的值。size_splits中的元素总和为维度split_dim的大小。调用说明调用方式样例代码说明aclnn接口test_aclnn_split_tensor通过aclnnSplitTensor接口方式调用SplitV算子。aclnn接口test_aclnn_split_with_size通过aclnnSplitWithSize接口方式调用SplitV算子。贡献说明贡献者贡献方贡献算子贡献时间贡献内容GMOW西北工业大学智能感知交互实验室SplitV2026/6/23SplitV算子适配开源仓【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考