edc

函数用法： 根据括号后面的维度将两个tensor拼接在一起，需要注意除了拼接的维度外，其他的维度需要相等才能完成匹配（代码里面设置的100） import torch x_0=torch.ones(1,100,100) y_0=torch.ones(2,100,100) z_0=torch.cat((x_0,y_0),0) x_1=torch.ones(100,1,100) y_1=torch.ones(100,2,100) z_1=torch.cat((x_1,y_1),1) x_2=torch.ones(100,100,1) y_2=torch.ones(100,100,2) z_2=torch.cat((x_2,y_2),2) print("dimension 0:",z_0.size()) print("dimension 1:",z_1.size()) print("dimension 2:",z_2.size()) 输出： dimension 0: torch.Size([3, 100, 100]) dimension 1: torch.Size([100, 3, 100]) dimension 2: torch.Size([100, 100, 3]) 来源： CSDN 作者： kking_edc 链接： https://blog.csdn.net/kking_edc