unsqueeze与squeeze函数
# squeeze(1)与squeeze(-1)的作用
两者效果一样,都是给tensor降维,但只能作用于n*1维度的tensor,调用后效果如图
但如果不是n*1的这种2维tensor,如本就是1维或m*n这种,调用该函数无任何效果
# squeeze(0)的作用
当张量是一个1*n维度的张量时,例如:张量[[1, 2, 3]]是一个1*3维的,调用这个函数后的效果图如下:
但是如果不是1*n的这种2维张量的话,如本就是1维的,或者m*n(其中m和n都是大于1的)这种的话,调用这个函数一点效果没有。
# unsqueeze(1)和unsqueeze(-1)的作用
和squeeze(1)和squeeze(-1)是反着的,如果我就是一个一维的张量,我调用这个函数,我就变成上面第1节中的原本的样子。
例如:张量[1.2, -5.6, 9, 0.004],调用这两个函数后的效果为:
如果我就是一n*m的2维的张量,调用这两个函数后的效果是啥呢?果然预期的一样,一点效果没有
例如:
# unsqueeze(0)的作用
和squeeze(0)的作用是反的
例如:张量[1.2, -5.6, 9, 0.004],调用这个函数后的效果为:
编辑 (opens new window)
上次更新: 2024/05/30, 07:49:34