can this be done with `np.reshape`, without `atleast_2d`?