torch.nn.functional.dropout2d

torch.nn.functional.dropout2d(input, p=0.5, training=True, inplace=False) [исходный код]

Случайным образом обнуляет целые каналы (канал — это двумерная карта признаков).

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

Подробности см. в Dropout2d.

Параметры:

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

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

Tensor