Ensure model output vector in unit vector