如何在pytorch中从图像中提取补丁？

import torch
import torchvision.transforms as transforms
from PIL import Image

image = Image.open('image.jpg')  # 替换为你的图像路径
preprocess = transforms.Compose([
    transforms.Resize((224, 224)),  # 调整图像大小
    transforms.ToTensor()  # 转换为张量
])
input_tensor = preprocess(image).unsqueeze(0)  # 添加批次维度

model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained=True)
model.eval()  # 设置为评估模式

output = model(input_tensor)
patch = output[0, :, 0, 0]  # 提取第一个补丁（示例中为1x1大小的补丁）