Split - 2 vs 18

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.

Files changed (1) hide show
  1. Split2 → Split18 +14 -7
Split2 → Split18 RENAMED
@@ -1 +1 @@
1
- Split a tensor into a list of tensors, along the specified
1
+ Split a tensor into a list of tensors, along the specified 'axis'.
2
- 'axis'. Lengths of the parts can be specified using argument 'split'.
2
+ Either input 'split' or the attribute 'num_outputs' should be specified, but not both.
3
- Otherwise, the tensor is split to equal sized parts.
3
+ If the attribute 'num_outputs' is specified, then the tensor is split into equal sized parts.
4
+ If the tensor is not evenly splittable into num_outputs, the last chunk will be smaller.
5
+ If the input 'split' is specified, it indicates the sizes of each output in the split.
4
6
  ### Attributes
5
7
  * **axis - INT** (default is '0'):
6
- Which axis to split on.
8
+ Which axis to split on. A negative value means counting dimensions from the back. Accepted range is [-rank, rank-1] where r = rank(input).
7
- * **split - INTS** :
9
+ * **num_outputs - INT** :
8
- length of each output
10
+ Number of outputs to split parts of the tensor into. If the tensor is not evenly splittable the last chunk will be smaller.
9
11
  ### Inputs
12
+
13
+ Between 1 and 2 inputs.
10
14
  - **input** (heterogeneous) - **T**:
11
15
  The tensor to split
16
+ - **split** (optional, heterogeneous) - **tensor(int64)**:
17
+
18
+ Optional length of each output. Values should be >= 0.Sum of the values must be equal to the dim value at 'axis' specified.
12
19
  ### Outputs
13
20
  Between 1 and 2147483647 outputs.
14
21
  - **outputs** (variadic, heterogeneous) - **T**:
15
22
  One or more outputs forming list of tensors after splitting
16
23
  ### Type Constraints
17
- * **T** in ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) ):
24
+ * **T** in ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) ):
18
25
  Constrain input and output types to all tensor types.