keras/src/trainers/data_adapters/array_data_adapter.py/RandomBatchSampler/__iter__
class RandomBatchSampler:            def __iter__(self):
                for batch in self.sampler:
                    yield [batch[i] for i in torch.randperm(len(batch))]