torch.unsqueeze()

torch.unsqueeze(input, dim, out=None)

返回一个新的张量，对输入的既定位置插入维度 1

 

torch.squeeze(input, dim=None, out=None)

将输入张量维度中的1 去除并返回。 如果输入是形如(A×1×B×1×C×1×D)，那么输出形状就为： (A×B×C×D)

当给定dim时，那么挤压操作只在给定维度上。例如，输入形状为: (A×1×B), squeeze(input, 0) 将会保持张量不变，只有用 squeeze(input, 1)，形状会变成 (A×B)。