So something like this would be fine? Would it make the data loader to be slower than normal?

def __init(self,data):
        # DO STUFF
 		batch_list = [[],[]]
		c_batches = []

def __getitem__(self,index):
        for data in index:
			result = imagenet_data[data]
			batch_list[0].append(result[0])
			batch_list[1].append(result[1])
		c_batches.append(batch_list)

        return c_batches #<------ This contain a batch