If you’ve created the patches in __getitem__, you could return all of them and create another loop over these patches inside your DataLoader loop, if necessary.
Alternatively, if you’ve created the patches before and store them e.g. in a list, you could simply return a single patch by indexing the container (list).