1# Split
2
3
4## 概述
5
6算子沿 axis 维度将x拆分成多个张量,张量数量由outputNum指定。
7
8该算子对应的[NodeType](_n_n_rt_v10.md#nodetype)为NODE_TYPE_SPLIT。
9
10输入:
11
12- x,n维tensor
13
14输出:
15
16- outputs,一组n维张量,每一个张量类型和维度相同,每个张量的类型和x一致。
17
18**起始版本:** 3.2
19
20**相关模块:**[NNRt](_n_n_rt_v10.md)
21
22
23## 汇总
24
25
26### Public 属性
27
28| 名称 | 描述 |
29| -------- | -------- |
30| long [outputNum](#outputnum) | 指定分割数量。 |
31| long[] [sizeSplits](#sizesplits) | 指定 输入的张量沿 axis 轴拆分后,每个张量的大小。 |
32| long [axis](#axis) | 指定分割的维度。 |
33
34
35## 类成员变量说明
36
37
38### axis
39
40```
41long Split::axis
42```
43
44**描述**
45
46
47指定分割的维度。
48
49
50### outputNum
51
52```
53long Split::outputNum
54```
55
56**描述**
57
58
59指定分割数量。
60
61
62### sizeSplits
63
64```
65long [] Split::sizeSplits
66```
67
68**描述**
69
70
71指定 输入的张量沿 axis 轴拆分后,每个张量的大小。
72
73- 如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的 张量,此时要求 x.shape[axis] 可以被 outputNum 整除;
74
75- 如果 sizeSplits 不为空,则要求 sizeSplits 所有元素之和等于 x.shape[axis]。
76