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