假设data为(3,64,64)的tensor向量,使用torch.unsqueeze将其变成(1,3,64,64)的向量,再用cat进行拼接,存储到change_pic变量中
isfirst=0
input_resize=torch.unsqueeze(data,0)
if isfirst==0:
change_pic=input_resize
isfirst+=1
else:
change_pic=torch.cat((change_pic,input_resize),0)