功能
对splitDim维进行splitNum等分。
图1 SplitOperation
约束
输入x的维度须能够被splitNum整除。
定义
struct SplitParam {
int32_t splitDim = 0;
int32_t splitNum = 2;
};
成员
成员名称 |
描述 |
|---|---|
splitDim |
需要进行拆分的维度对应的索引。 splitDim须位于输入x的维度范围内，即如果x的维度为xDim，则splitDim的取值范围为[-xDim, xDim - 1]。 当splitDim为负数时，其含义是从最高维度开始访问，如splitDim = -1，x维度数为dimNum，则拆分维度为dimNum - 1。 |
splitNum |
Split的等分次数，当前支持2或3。 splitNum = 3时输入x要求是float16数据类型。 |
输入
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/int64/bf16 |
ND |
最高8维。 |
输出
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
z |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/int64/bf16 |
ND |
- |
z1 |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/int64/bf16 |
ND |
- |
z2 |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/bf16 |
ND |
splitNum=3时才有z2，仅float16支持。 |