CODE EXAMPLE FOR PYTHON

numpy split to chunks of equal size

  def build_chunks(arr, chunk_size, axis=1):
     return np.split(arr, 
                     range(chunk_size, arr.shape[axis], chunk_size), axis=axis)