实习记录(2)

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

举例:

 unsqueeze是和squeeze相反的操作

 

posted @ 2023-07-04 19:38  Halo辉Go  阅读(2)  评论(0编辑  收藏  举报