 it's maybe np.sum(axis=1)