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