降维与升维
1.tensor.squeeze() 降维
(1)若squeeze()括号内为空,则将张量中所有维度为1的维数进行压缩,如将2131的张量降维到23维;若维度中无1维的维数,则保持源维度不变,如将234维的张量进行squeeze,则转换后维度不会变。
(2)若squeeze(idx),则将张量中对应的第idx维de的维度进行压缩,如2131的张量做squeeze(1),则会降维到23*1维的张量;若第idx维度的维数不为1,则squeeze后维度不会变化。
2.tensor.unsqueeze(idx)升维
在第idx维进行升维,将tensor由原本的维度n,升维至n+1维。如张量的维度维23,经unsqueeze(1)后,变为21*3维度的张量