The array_split package provides the means to partition an array (or array shape) using any of the following criteria: Per-axis indices indicating the cut positions. Per-axis number of sub-arrays.