tensor.squeeze(index) 字面意思:就是在index处,如果维度为1,就删掉这个维度。如果此处维度不为1,则不操作
tensor.unsqueeeze(index) :就是在index处,增加一个维度为1的维度;
注意 比如tensor a 的维度是(2,1,3)
a.squeeze(1) 的维度是(2,3),但a的维度仍然是(2,1,3)。就是squeeze和unsqueeze不会对tensor本身产生修改,修改后要赋值给别的变量才能够保存。
如 b = a.squeeze(1)