Flatten

class torch.nn.modules.flatten.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])

extra_repr()

返回模組的額外表示。

返回型別

str

forward(input)

執行前向傳播。

返回型別

張量