评价此页

Flatten#

class torch.nn.modules.flatten.Flatten(start_dim=1, end_dim=-1)[source]#

将连续的维度范围展平成一个张量。

用于与 Sequential 结合使用,有关详细信息,请参阅 torch.flatten()

形状
  • 输入: (,Sstart,...,Si,...,Send,)(*, S_{\text{start}},..., S_{i}, ..., S_{\text{end}}, *),其中 SiS_{i} 是维度 ii 的大小,而 * 表示任意数量的维度,包括零个。

  • 输出: (,i=startendSi,)(*, \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()[source]#

返回模块的额外表示。

返回类型

str

forward(input)[source]#

执行前向传播。

返回类型

张量