在PyTorch中提取图像的编码表示？

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

# 加载预训练的ResNet模型
resnet = models.resnet50(pretrained=True)
# 设置模型为评估模式
resnet.eval()

# 图像预处理
transform = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])

# 加载图像
image = Image.open('image.jpg')
# 进行预处理
image = transform(image)
# 添加一个维度以匹配模型输入的形状
image = image.unsqueeze(0)

# 使用ResNet模型提取图像的编码表示
encoding = resnet(image)

# 打印编码表示的形状
print(encoding.shape)