如何使用Pytorch DataLoader输出x和y值的较大2D网格的小2D块？

import torch
from torch.utils.data import Dataset, DataLoader

class GridDataset(Dataset):
    def __init__(self, grid_size):
        self.grid_size = grid_size

    def __len__(self):
        return self.grid_size ** 2

    def __getitem__(self, idx):
        x = idx // self.grid_size
        y = idx % self.grid_size
        return x, y

grid_size = 10  # 定义网格大小
dataset = GridDataset(grid_size)
dataloader = DataLoader(dataset, batch_size=4, shuffle=True)

for batch in dataloader:
    x, y = batch
    # 在这里进行对小2D块的处理
    print(x, y)