The input array to split.
If an integer, it indicates the number of equal sections to create along the specified axis. If a list of integers, it specifies the indices at which to split the array.
The axis along which to split the array. Default is 0.
Split an array into multiple sub-arrays along an axis.