实习记录(2)

发布时间 2023-07-04 19:46:50作者: Halo辉Go

torch中的squeeze和unsqueeze
squeeze是压缩,对维度进行降维。不写的话,默认将所有维度为1的去掉(我理解就是去掉对应层的"[]"中括号)

举例:

 unsqueeze是和squeeze相反的操作