PyTorch flatten() 和 squeeze() 区别

在 PyTorch 里，flatten() 和 squeeze(0) 是两个不同的张量操作，

1. flatten() 方法

flatten() 方法用于把一个多维张量展开成一维张量。它会将张量里的所有元素按顺序排列成一个一维序列。

语法

torch.flatten(input, start_dim=