pytorch中tensor的squeeze()和unsqueeze()
squeeze():
squeeze(arg)表示第arg维的维度值为1,则去掉该维度,否则tensor不变。
c = torch.randn(5,1)
c
tensor([[ 0.7312],
[ 1.5407],
[ 0.4843],
[ 0.0443],
[-0.8765]])
c.shape
结果:torch.Size([5, 1])
c.squeeze()
结果:tensor([ 0.7312, 1.5407, 0.4843, 0.0443, -0.8765])
unsqueeze():
unsqueeze(arg)与squeeze(arg)作用相反,表示在第arg维增加一个维度值为1的维度。
在这里插入代码片