Spec-Zone.ru › PyTorch 1

torch.nn.functional.feature_alpha_dropout

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

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

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

См. FeatureAlphaDropout для подробностей.

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

Тензор

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.nn.functional.feature_alpha_dropout.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API