torch.nn.functional.dropout1d

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

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

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

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

Параметры:

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

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

Tensor