多个tensor合并成一个——四个三维tensor合成一个四维tensor——四个[3,512,1024]变成[4,3,512,1024]
仅作为记录,大佬请跳过。四个[3,512,1024]变成[4,3,512,1024]:——先将[3,512,1024]用torch.unsqueeze(),变成[1,3,512,1024]——用for循环,分别对四个三维tensor做unsqueeze——最后用torch.cat((a,b),0)(torch.cat可以对多个tensor进行合并:torch.cat((a,b,c,d),0);注: