squeeze和unsqueeze

squeeze

squeeze(x, axis=None, name=None)
如果指定了axis,那么会删除axis中指定尺寸为1的维度(如果axis指定的尺寸不为1,则不会删除该维度

  1. a.shape=[1,2,3],假设要删除第0个维度,那么可以直接squeeze(0),注意此时第0维度为1,符合删除的要求。
    squeeze和unsqueeze_第1张图片
  2. 当不符合维度不符合要求的时候,则保持不变。
    squeeze和unsqueeze_第2张图片

unsqueeze

unsqueeze(x, axis, name=None)
插入一个尺度为1的维度,其中axis表示新插入的维度在新shape 的第几个位置。
squeeze和unsqueeze_第3张图片

你可能感兴趣的:(pytorch,paddlepaddle,tensorflow)