torch.nn.functional.feature_alpha_dropout

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

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

Каждый элемент независимо маскируется на каждом вызове forward с вероятностью p используя выборки из распределения Бернулли. Маскируемые элементы случайны при каждом вызове forward и масштабируются и сдвигаются, чтобы сохранить среднее значение 0 и дисперсию 1.

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

Параметры

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

Возвращаемый тип

Тензор