torch.nn.functional.dropout2d

torch.nn.functional.dropout2d(input, p=0.5, training=True, inplace=False) [source]

Случайным образом обнуляет целые каналы (канал — это 2D карта признаков, например, $j$-й канал $i$-й выборки в пакетном вводе — это 2D тензор $\text{input}[i, j]$) входного тензора). Каждый канал будет независимо обнуляться на каждом вызове прямого вычисления с вероятностью p с использованием выборок из биномиального распределения.

См. Dropout2d для получения подробной информации.

Параметры:

• p (float) – вероятность обнуления канала. Значение по умолчанию: 0.5
• training (bool) – применять дропаут, если True. Значение по умолчанию: True
• inplace (bool) – Если установлено True, то данная операция выполняется in-place. Значение по умолчанию: False

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

Тензор