squeeze可以将张量中尺寸为1的维度移除。
也可以指定维度。
注意：返回的张量与原张量共享内存，改变一个会同时改变另一个。
Example:
>>> x = torch.zeros(2, 1, 2, 1, 2) >>> x.size() torch.Size([2, 1, 2, 1, 2]) >>> y = torch.squeeze(x) >>> y.size() torch.Size([2, 2, 2]) >>> y = torch.squeeze(x, 0) >>> y.size() torch.Size([2, 1, 2, 1, 2]) >>> y = torch.squeeze(x, 1) >>> y.size() torch.Size([2, 2, 1, 2]) 转载于:https://www.cnblogs.com/jiangkejie/p/10310299.html