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, операция будет выполняться на месте. Значение по умолчанию: False

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

Тензор