changing computation of batch slices to np.array_split