I have built a GAN model with Keras and converted the generator to a mlmodel and I\'m trying to use it in my iOS app. The output of the model is a MLMultiArray of shape (1,