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) 

Numpy split array into chunks of equal size

def split_given_size(a, size):
    return np.split(a, np.arange(size,len(a),size))