Split - 1 vs 2

Files changed (1) hide show
  1. Split1 → Split2 +18 -11
Split1 → Split2 RENAMED
@@ -1 +1 @@
1
1
  Split a tensor into a list of tensors, along the specified
2
- 'axis'. The lengths of the split can be specified using argument 'axis' or
2
+ 'axis'. Lengths of the parts can be specified using argument 'split'.
3
+ Otherwise, the tensor is split to equal sized parts.
3
- optional second input blob to the operator. Otherwise, the tensor is split
4
- to equal sized parts.
5
4
  **Attributes**
6
5
  * **axis**:
7
- Which axis to split on
6
+ Which axis to split on.
8
7
  * **split**:
9
8
  length of each output
10
9
  **Inputs**
11
- Between 1 and 2 inputs.
12
-
13
10
  * **input** (heterogeneous) - **T**:
14
11
  The tensor to split
15
- * **split** (optional, heterogeneous) - **T**:
16
- Optional list of output lengths (see also arg 'split')
17
12
  **Outputs**
18
13
  Between 1 and 2147483647 outputs.
19
- * **outputs...** (variadic, heterogeneous) - **T**:
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 (
18
+ tensor(bool),
19
+ tensor(complex128),
20
+ tensor(complex64),
23
21
  tensor(double),
24
22
  tensor(float),
25
- tensor(float16)
23
+ tensor(float16),
24
+ tensor(int16),
25
+ tensor(int32),
26
+ tensor(int64),
27
+ tensor(int8),
28
+ tensor(string),
29
+ tensor(uint16),
30
+ tensor(uint32),
31
+ tensor(uint64),
32
+ tensor(uint8)
26
33
  ):
27
- Constrain input types to float tensors.? ^ ^^^
34
+ Constrain input and output types to all tensor types.? +++++++++++ ^ ^ +++++