torch.cat（）用法

torch.cat()是用来拼接tensor的

语法：

torch.cat（[x1,x2,...],dim = )

x1,x2为要拼接的矩阵，dim为你指定拼接的维度，

x = torch.randn(2,3,4)
y = torch.randn(2,3,4)
z = []

for i in range(len(x)+1):
    print('dim=',i)
    z = torch.cat([y,x],i)
    print(z.shape)

指定的维度不能超过要拼接矩阵本身的维度

除了在你指定的维度上，要拼接的矩阵在其他维度的形状大小要相同

x = torch.randn(2,3,3)
y = torch.randn(3,3,3)
z= torch.cat([y,x],0)
print(z.shape)

# torch.Size([5, 3, 3])

如果指定拼接维度为1，但是在维度0，两拼接矩阵大小不一致，就会报错

x = torch.randn(2,3,3)
y = torch.randn(3,3,3)
z= torch.cat([y,x],1)
print(z.shape)