文件最后提交记录最后更新时间
1 个月前
1 个月前
2 个月前
2 个月前
2 个月前
2 个月前
2 个月前
2 个月前
README

SplitV

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

  • 算子功能:根据size_splits将张量沿维度split_dim拆分为num_split更小的张量。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 需要切分的tensor列表。 BOOL、INT8、UINT8、BFLOAT16、FLOAT16、INT16、UINT16、FLOAT32、INT32、UINT32、INT64、UINT64 ND
size_splits 输入 指定一个列表,其中包含沿分割维度的每个输出张量的大小。 INT64 ND
split_dim 输入 指定沿其分割的维度。 INT64 ND
num_split 属性 指定要分割的tensor个数。 INT -
y 输出 输出结果。 BOOL、INT8、UINT8、BFLOAT16、FLOAT16、INT16、UINT16、FLOAT32、INT32、UINT32、INT64、UINT64 ND

约束说明

  • 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 西北工业大学智能感知交互实验室 SplitV 2026/6/23 SplitV算子适配开源仓