torch.nn.functional.dropout3d

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

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

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

Подробности см. в описании Dropout3d.

Параметры:

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

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

Tensor