x = x / np.sum(x, axis=-1)[...,np.newaxis]