图片tensor改变为随意大小

import torch as t
from Data import DataLayer
from skimage.transform import resize
from torchvision.utils import save_image

dataset = DataLayer(data_path=data_path, current_fold=int(current_fold), organ_number=organ_number, 
		low_range=low_range, high_range=high_range, slice_threshold=slice_threshold, slice_thickness=slice_thickness, 
		organ_ID=organ_ID, plane=plane)
dataloader = t.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=True, num_workers=16, drop_last=True)
data_iter = iter(dataloader)
data = next(data_iter)
ven, art, label = data
ven1 = ven[:, 1:2, :, :]
print(ven1.shape)
save_image(ven1, 'a.png')
ven2=t.from_numpy(resize(ven1.cpu().numpy(), (1,3,64,64)))
print(ven2.shape)
save_image(ven2, 'b.png')