Split - 11 vs 13

Files changed (1) hide show
  1. Split11 → Split13 +7 -3
Split11 → 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 argument 'split'.
2
+ 'axis'. Lengths of the parts can be specified using input 'split'.
3
3
  Otherwise, the tensor is split to equal sized parts.
4
4
  **Attributes**
5
5
  * **axis**:
6
6
  Which axis to split on. A negative value means counting dimensions
7
7
  from the back. Accepted range is [-rank, rank-1] where r =
8
8
  rank(input).
9
- * **split**:
10
- length of each output. Values should be >= 0.
11
9
  **Inputs**
10
+ Between 1 and 2 inputs.
11
+
12
12
  * **input** (heterogeneous) - **T**:
13
13
  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.
14
17
  **Outputs**
15
18
  Between 1 and 2147483647 outputs.
16
19
  * **outputs** (variadic, heterogeneous) - **T**:
17
20
  One or more outputs forming list of tensors after splitting
18
21
  **Type Constraints**
19
22
  * **T** in (
23
+ tensor(bfloat16),
20
24
  tensor(bool),
21
25
  tensor(complex128),
22
26
  tensor(complex64),
23
27
  tensor(double),
24
28
  tensor(float),
25
29
  tensor(float16),
26
30
  tensor(int16),
27
31
  tensor(int32),
28
32
  tensor(int64),
29
33
  tensor(int8),
30
34
  tensor(string),
31
35
  tensor(uint16),
32
36
  tensor(uint32),
33
37
  tensor(uint64),
34
38
  tensor(uint8)
35
39
  ):
36
40
  Constrain input and output types to all tensor types.