the input tensor
either an integer indicating the number of splits along the axis or an array of integers containing the sizes of each output tensor along the axis. If a number then it must evenly divide the axis length; otherwise the sum of sizes must match axis length.
Optional axis: number(optional) the dimension along which to split
Generated using TypeDoc
Splits a tensor into sub tensors.
Returns
the split tensor