torch.squeeze

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

>>阅读原文<<