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