The input
Number of splits. Note that this should evenly divide the length of the axis.
Axis along which to split.
If true, Removes the axis with length 1 from the shapes of the output arrays. **Note** that setting squeeze_axis to removes axis with length 1 only along the trueaxis which it is split. Also squeeze_axis can be set to only if true.
input.shape[axis] == num_outputs
This Param Object is specifically used for split