Split - 2 vs 13#
Next section compares an older to a newer version of the same operator after both definition are converted into markdown text. Green means an addition to the newer version, red means a deletion. Anything else is unchanged.
- Split2 → Split13 +4 -10
Split2 → Split13
RENAMED
@@ -1 +1 @@
|
|
1
1
|
Split a tensor into a list of tensors, along the specified
|
2
|
-
'axis'. Lengths of the parts can be specified using
|
2
|
+
'axis'. Lengths of the parts can be specified using argument 'split'.
|
3
3
|
Otherwise, the tensor is split to equal sized parts.
|
4
4
|
**Attributes**
|
5
5
|
* **axis**:
|
6
|
-
Which axis to split on.
|
6
|
+
Which axis to split on.
|
7
|
-
from the back. Accepted range is [-rank, rank-1] where r =
|
8
|
-
|
7
|
+
* **split**:
|
8
|
+
length of each output
|
9
9
|
**Inputs**
|
10
|
-
Between 1 and 2 inputs.
|
11
|
-
|
12
10
|
* **input** (heterogeneous) - **T**:
|
13
11
|
The tensor to split
|
14
|
-
* **split** (optional, heterogeneous) - **tensor(int64)**:
|
15
|
-
Optional length of each output. Values should be >= 0.Sum of the
|
16
|
-
values must be equal to the dim value at 'axis' specified.
|
17
12
|
**Outputs**
|
18
13
|
Between 1 and 2147483647 outputs.
|
19
14
|
* **outputs** (variadic, heterogeneous) - **T**:
|
20
15
|
One or more outputs forming list of tensors after splitting
|
21
16
|
**Type Constraints**
|
22
17
|
* **T** in (
|
23
|
-
tensor(bfloat16),
|
24
18
|
tensor(bool),
|
25
19
|
tensor(complex128),
|
26
20
|
tensor(complex64),
|
27
21
|
tensor(double),
|
28
22
|
tensor(float),
|
29
23
|
tensor(float16),
|
30
24
|
tensor(int16),
|
31
25
|
tensor(int32),
|
32
26
|
tensor(int64),
|
33
27
|
tensor(int8),
|
34
28
|
tensor(string),
|
35
29
|
tensor(uint16),
|
36
30
|
tensor(uint32),
|
37
31
|
tensor(uint64),
|
38
32
|
tensor(uint8)
|
39
33
|
):
|
40
34
|
Constrain input and output types to all tensor types.
|