如何将两个维度为(3,64,64)的tensor向量拼接成(2,3,64,64)的tensor向量？

假设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)