My following code can run for some batches but at the end of the first episode, it shows the following errors.
The generator generates 2d arrays for x and 1d arrays f
