Would this be `image[..., np.newaxis]`, assuming the final dimension denotes channels?