【Pytorch】Tensor维度的拼接与拆分

1. 拼接与拆分常用API

• cat函数
• stack函数
• split函数
• chunk函数

2. 按照维度合并Tensor

2.1 cat函数

def cat(tensors, dim) -> Tensor

• tensors：需要合并的Tensor
• dim：按照维度dim进行合并
• 注意：想要拼接的维度上的值可以不同，但是其它维度上的值必须相同，并且两个 Tensor 的维度最大值必须相同

a = torch.rand(4, 32, 8)  # 含义： 4个班级，每个班级32个人，每个人8门课成绩
b = torch.rand(5, 32, 8)  # 含义： 5个班级，每个班级32个人，每个人8门课成绩

# 合并成绩单
c = torch.cat([a,b], dim=0)
print(c.shape)     #  torch.Size([9, 32, 8])	-> 9个班级，每个班级32个人，每个人8门课成绩

解释：从班级维度( 0D )将成绩进行合并，两个 Tensor 中的 0D 中的值可以不同，但是其他维度上的值必须相同

二维矩阵理解 cat 函数