Regression via Kirszbraun Extension with Applications to Imitation Learning