Dropout2d

class torch.nn.modules.dropout.Dropout2d(p=0.5, inplace=False) [source]

Случайным образом обнуляет целые каналы.

Канал — это двумерная карта признаков, например $j$-й канал $i$-го элемента пакетного входного тензора представляет собой двумерный тензор $\text{input}[i, j]$.

При каждом прямом проходе каждый канал независимо обнуляется с вероятностью p с использованием выборок из распределения Бернулли.

Обычно входные данные поступают из модулей nn.Conv2d.

Как описано в статье Эффективная локализация объектов с помощью свёрточных сетей , если соседние пиксели внутри карт признаков сильно коррелируют (что обычно происходит в ранних свёрточных слоях), то независимое прореживание не будет регуляризовать активации и приведёт лишь к фактическому снижению скорости обучения.

В этом случае nn.Dropout2d() поможет обеспечить независимость между картами признаков, и его следует использовать вместо этого.

Параметры:

• p (float, необязательный) – вероятность обнуления элемента.
• inplace (bool, необязательный) – Если задано значение True, операция выполняется на месте

Предупреждение

По историческим причинам этот класс выполняет прореживание по каналам в одномерном режиме для трёхмерных входных данных (как это делает nn.Dropout1d). Поэтому в настоящее время он НЕ поддерживает входные данные без измерения пакета с формой $(C, H, W)$. В будущей версии поведение изменится: трёхмерные входные данные будут интерпретироваться как входные данные без измерения пакета. Чтобы сохранить прежнее поведение, переключитесь на nn.Dropout1d.

Форма:

• Вход: $(N, C, H, W)$ или $(N, C, L)$.
• Выход: $(N, C, H, W)$ или $(N, C, L)$ (та же форма, что и у входных данных).

Примеры:

>>> m = nn.Dropout2d(p=0.2)
>>> input = torch.randn(20, 16, 32, 32)
>>> output = m(input)

forward(input) [source]

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

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

Tensor