矩阵合并

1、 使用cat函数

import torch
a = torch.zeros((2,4))
b = torch.ones((2,4))

out = torch.cat((a,b),dim = 0)
print(out)

dim = 0的时候按照纵轴进行拼接行数

dim = 1时候，按照横轴镜像列数拼接

2、 使用stack
stack是把两个维度进行独立拼接，形成一个新的维度。这个用的相对较少。

3、tensor的切片
按照某个维度平均分块，或者依照第二个参数给出的list进行分割。
dim是根据列来切（横轴不变，纵轴切分）

a = torch.rand((3,4))
print(a)
out = torch.chunk(a,2,dim=1)
print(out[0],out[0].shape)
print(out[0],out[0].shape)

3×4的矩阵分为 3×2和 3×2