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