Set Prediction in the Latent Space