Dropout1d
-
class torch.nn.Dropout1d(p=0.5, inplace=False)[source] -
Случайно обнуляет целые каналы (канал — это 1D карта признаков, например, -й канал -й выборки в пакетном входе представляет собой 1D тензор ). Каждый канал будет обнуляться независимо на каждом вызове с вероятностью
pс использованием выборок из биномиального распределения.Обычно вход поступает из
nn.Conv1dмодулей.Как описано в статье Efficient Object Localization Using Convolutional Networks, если смежные пиксели в картах признаков сильно коррелированы (как обычно бывает в ранних сверточных слоях), то независимое дропаут не будет регулировать активации и в противном случае просто приведет к уменьшению эффективной скорости обучения.
В этом случае,
nn.Dropout1d()поможет способствовать независимости между картами признаков и должен использоваться вместо него.- Параметры:
- Форма:
-
- Вход: или .
- Выход: или (такая же форма, как и вход).
Примеры:
>>> m = nn.Dropout1d(p=0.2) >>> input = torch.randn(20, 16, 32) >>> output = m(input)
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.nn.Dropout1d.html