``` a[:, np.newaxis] ``` seems to be much faster
seems to be much faster