多个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)；注： 里面要加括号）

代码

for tensor_unsqueeze_num in range(4):
    inputs_resize = data_transforms[transform_state[ii]](inputs_tensor_Image[tensor_unsqueeze_num])
    inputs_resize_unsqueeze = torch.unsqueeze(inputs_resize, dim=0)
    if tensor_unsqueeze_num==0:
        _inputs=inputs_resize_unsqueeze
    else:
        _inputs=torch.cat((_inputs,inputs_resize_unsqueeze),0)

展示

参考

关于torch.cat

传送门1

传送门2