Are you proposing that the dataset always should produce a single sample x, y for each call to getitem?
If so, how does one augment the dataset so that it incorporates random rotations/crops/shifts like here:
https://keras.io/preprocessing/image/ 30
?
The only solution I can see is that it will randomly select a sample and then randomly select a transformation, and then produce a single sample