Split - 2 vs 11

Files changed (1) hide show
  1. Split2 → Split11 +4 -2
Split2 → Split11 RENAMED
@@ -1 +1 @@
1
1
  Split a tensor into a list of tensors, along the specified
2
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. A negative value means counting dimensions
7
+ from the back. Accepted range is [-rank, rank-1] where r =
8
+ rank(input).
7
9
  * **split**:
8
- length of each output
10
+ length of each output. Values should be >= 0.
9
11
  **Inputs**
10
12
  * **input** (heterogeneous) - **T**:
11
13
  The tensor to split
12
14
  **Outputs**
13
15
  Between 1 and 2147483647 outputs.
14
16
  * **outputs** (variadic, heterogeneous) - **T**:
15
17
  One or more outputs forming list of tensors after splitting
16
18
  **Type Constraints**
17
19
  * **T** in (
18
20
  tensor(bool),
19
21
  tensor(complex128),
20
22
  tensor(complex64),
21
23
  tensor(double),
22
24
  tensor(float),
23
25
  tensor(float16),
24
26
  tensor(int16),
25
27
  tensor(int32),
26
28
  tensor(int64),
27
29
  tensor(int8),
28
30
  tensor(string),
29
31
  tensor(uint16),
30
32
  tensor(uint32),
31
33
  tensor(uint64),
32
34
  tensor(uint8)
33
35
  ):
34
36
  Constrain input and output types to all tensor types.