1. 拼接
1.1 cat()
torch.cat([tensor1,tensor2],dim)
[tensor1,tensor2]表示需要拼接的张量;
dim 表示在哪个维度上拼接
注意:拼接时,除了要拼接的维度外,其他维度的形状必须相同。
示意图:根据不同的维度拼接
1.2 stack()
stack在拼接张量时,会创建新的维度。
1.3 cat() VS stack()
2. 拆分
2.1 split()
.split(len,dim=0) # 按固定长度将张量拆分,dim表示在哪个维度上拆分,拆分后的每个张量在维度dim上的形状都是len。
.split([len1,len2,...],dim=0) # 按[len1,len2,...]中的长度将张量拆分为dim维度上形状不固定的若干个张量,dim表示在哪个维度上拆分。
2.2 chunk()
.chunk(num,dim) # 按数量拆分,表示张量按dim维度分成num块。