admin管理员组

文章数量:1123274

多个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_unsqueezeelse:_inputs=torch.cat((_inputs,inputs_resize_unsqueeze),0)

展示

参考

关于torch.cat

传送门1

传送门2

本文标签： 多个tensor合并成一个四个三维tensor合成一个四维tensor四个35121024变成431024

版权声明：本文标题：多个tensor合并成一个——四个三维tensor合成一个四维tensor——四个[3,512,1024]变成[4,3,512,1024] 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.betaflare.com/biancheng/1708457458a744479.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。