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

>>阅读原文<<