Flatten¶
- class torch.nn.Flatten(start_dim=1, end_dim=-1)[源代码]¶
将连续的维度范围展平为一个张量。
有关在
Sequential
中的使用,请参阅torch.flatten()
以获取详细信息。- 形状
输入:, 其中 是维度 上的大小,而 表示任意数量的维度,包括无维度。
输出:.
- 示例:
>>> 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])