Flatten

class torch.nn.Flatten(start_dim=1, end_dim=-1)

将连续维度范围展平为张量。

与 Sequential 一起使用，详情请参阅 torch.flatten()。

形状

• 输入: $(*, S_{\text{start}},..., S_{i}, ..., S_{\text{end}}, *)$,’ 其中 $S_{i}$ 是维度 $i$ 的大小，$*$ 表示任意数量的维度，包括无维度。

• 输出: $(*, \prod_{i=\text{start}}^{\text{end}} S_{i}, *)$。

参数

• start_dim (int) – 要展平的第一个维度 (默认值 = 1)。

• end_dim (int) – 要展平的最后一个维度 (默认值 = -1)。

示例：

>>> input = torch.randn(32, 1, 5, 5)
>>> # With default parameters
>>> m = nn.Flatten()
>>> output = m(input)
>>> output.size()
torch.Size([32, 25])
>>> # With non-default parameters
>>> m = nn.Flatten(0, 2)
>>> output = m(input)
>>> output.size()
torch.Size([160, 5])