向量维度降维,比如把(1,600)的向量T,它的第一维度值是1,第二维度值是600。通过T.squeeze(0)降维为(600,)
- squeeze(0)代表若第一维度值为1则去除第一维度
- squeeze(1)代表若第二维度值为1则去除第二维度
- -1,去除最后维度值为1的维度
学习来源于Python-squeeze()、unsqueeze()函数的理解_python squeeze-CSDN博客
向量维度降维,比如把(1,600)的向量T,它的第一维度值是1,第二维度值是600。通过T.squeeze(0)降维为(600,)
学习来源于Python-squeeze()、unsqueeze()函数的理解_python squeeze-CSDN博客