如何使用torchvision.io.read_image的图像作为变量，而不是存储的文件？

import torch
import torchvision.io as io
from PIL import Image
from io import BytesIO

# 读取图像文件并转换为张量
image_path = 'path_to_image.jpg'
image_tensor = io.read_image(image_path)

# 将图像张量转换为PIL图像对象
pil_image = Image.fromarray(image_tensor.permute(1, 2, 0).numpy())

# 将PIL图像对象转换为字节流
byte_stream = BytesIO()
pil_image.save(byte_stream, format='JPEG')

# 从字节流中读取图像并转换为张量
byte_stream.seek(0)
image_tensor_variable = io.read_image(byte_stream)

# 打印图像张量的形状
print(image_tensor_variable.shape)