torch.cat

函数用法：
根据括号后面的维度将两个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/article/details/104728356