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/2.1/generated/torch.nn.Dropout1d.html