I\'m trying to develop a variational auto-encoder and test it on the iris dataset. I\'m trying to generate a 15x4 matrix after training my model on the dataset.