深度学习与PyTorch笔记8

Broadcast自动扩展

作用：expand维度扩展without copying data不需要复制数据。
步骤：
1、insert 1 dim ahead，最前面自动扩展一个维度，shape为1。
2、expand dim with size 1 to same size，将维度为1的扩展成与原数据相同的大小
feature maps:[4,32,14,14]
bias:[32,1,1]–[1,32,1,1]–[4,32,14,14]
省内存消耗。
默认从最后一个维度开始匹配。
没理解透=_=

拼接与拆分

cat

拼接
成绩单一到四班[class1-4,students,scores]
成绩单五到九班[class5-9,students,scores]
合并成绩单
torch.cat(list,dim=d)，第一个参数为list，包含所有需要统计的tensor，第二个参数为维度，在第d个维度上合并。

叠加时其余维度size必须相同。

stack

拼接，创建一个新的维度。原tensor的维度必须完全一致。新创建的维度的大小跟拼接的数据的个数有关。

split

按长度拆分，a=b.split(list,dim=d)，list中为拆分的每一段的长度。每段长度一样的话可以只给一个值a=b.split(2,dim=d)。

chunk

按数量拆分，a=b.chunk(list,dim=d)，list中为要拆分出来的数量，平均拆分。