torch.nn.functional.dropout

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

Во время обучения случайным образом обнуляет некоторые элементы входного тензора с вероятностью p, используя выборки из биномиального распределения.

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

Параметры

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

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

Тензор