keras/src/trainers/data_adapters/torch_data_loader_adapter.py/TorchDataLoaderAdapter/get_numpy_iterator
class TorchDataLoaderAdapter:    def get_numpy_iterator(self):
        for batch in self._dataloader:
            # shared memory using `np.asarray`
            yield tuple(
                tree.map_structure(lambda x: np.asarray(x.cpu()), batch)
            )