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) [исходный код]

Выполняет прямой проход.

Тип возвращаемого значения:

Tensor