把单张图片转换成张量形式

img1 = cv2.imread('OCT3.png') #读入一张图像
img1 = cv2.resize(img1,(256,256)) #修改尺寸为256*256
img1 = np.array(img1)
tensor = transforms.ToTensor()
img_tensor = tensor(img1) #转化成张量形式，此时的shape是（3，256，256）
img_tensor1 = img_tensor.unsqueeze(0) #有时候需要四维张量运算，把三维变成四维，此时的shape是（1，3，256，256）